pandas 笔记 stack unstack

原创已于 2023-07-20 14:59:02 修改 · 458 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pandas #笔记

于 2023-07-20 14:25:33 首次发布

python库整理专栏收录该内容

349 篇文章

订阅专栏

文章介绍了Pandas库中DataFrame的stack和unstack函数，用于行列转换。stack将列转换为行，形成multi-index；unstack则相反，将行转换回列。示例展示了如何使用这两个函数，并特别讨论了包含multi-index的情况以及设置unstack的level参数。

stack()：从列到行堆叠，把列转换为行
unstack()：从行到列展开，把行转换为列

1 stack

data=pd.DataFrame(np.arange(12).reshape((3,4)),
 index=pd.Index(['date1','date2','date3']),
 columns=pd.Index(['store1','store2','store3','store4'])
 )

data

data.stack().to_frame()

2 unstack

第一列index作为unstack后的index

为了和stack以示区别，我们交换前面stack后的两个index

data.stack().to_frame().swaplevel(0,1)

data.stack().swaplevel(0,1).unstack()

2.1 unstack 的内容是multi-index

2.1.1 数据

data=pd.DataFrame(np.arange(12).reshape((3,4)),
 index=[['date1','date1','date3'],
                ['idx1','idx2','idx1'],
       ['type1','type1','type2']],
 columns=['store1','store2','store3','store4']
 )

data