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

时间:2021-05-22

import numpy as npimport pandas as pdfrom pandas import Series,DataFrame

一、重塑

  • stack:将数据的列索引旋转为行索引
  • unstack:将数据的行索引旋转为列索引
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

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

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章