Pandas 重塑(stack)和轴向旋转(pivot)的实现

脚本专栏 2024/9/23 佚名

3 1 2

import numpy as np
import pandas as pd
from pandas import Series,DataFrame

一、重塑

df = DataFrame({'水果':['苹果','梨','草莓'],
        '数量':[3,4,5],
        '价格':[4,5,6]})
print(df)

   价格数量水果
0   4   3 苹果
1   5   4   梨
2   6   5 草莓

1.stack()

stack_df = df.stack()
print(stack_df)

0 价格     4
   数量     3
   水果    苹果
1 价格     5
   数量     4
   水果     梨
2 价格     6
   数量     5
   水果    草莓

dtype: object

2.unstack()

print(stack_df.unstack())

价格数量水果
0 4 3 苹果
1 5 4 梨
2 6 5 草莓

3.通过level参数指定旋转轴的层次(默认level=-1)

print(stack_df.unstack(level=0))

     0 1   2
价格   4 5   6
数量   3 4   5
水果苹果梨草莓

二、轴向旋转(pivot)

pivot(index,columns,values)：将index指定为行索引，columns是列索引，values则是DataFrame中的值

df = DataFrame({'水果种类':['苹果','苹果','梨','梨','草莓','草莓'],
        '信息':['价格','数量','价格','数量','价格','数量'],
        '值':[4,3,5,4,6,5]})
print(df)

   信息值水果种类
0 价格 4   苹果
1 数量 3   苹果
2 价格 5    梨
3 数量 4    梨
4 价格 6   草莓
5 数量 5   草莓

将水果种类作为行索引，将信息作为列索引

print(df.pivot('水果种类','信息','值'))

信息    价格数量
水果种类
梨      5   4
苹果     4   3
草莓     6   5

pivot可以用set_index和unstack等价的实现

print(df.set_index(['水果种类','信息']).unstack())

      值
信息   价格数量
水果种类
梨     5 4
苹果    4 3
草莓    6 5

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。

Pandas,重塑,Pandas,轴向旋转

华山资源网 Design By www.eoogi.com

广告合作：本站广告合作请联系QQ：858582 申请时备注：广告合作（否则不回）
免责声明：本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除！

华山资源网 Design By www.eoogi.com

评论“Pandas 重塑(stack)和轴向旋转(pivot)的实现”

暂无评论...

www.eoogi.com 华山资源网

120,135影音资源

344,641技术资源

22,817软件资源

435,032站长资源

2024年09月23日