Python中stack和unstack函数（附加reset_index,set_index函数）

最新推荐文章于 2024-11-24 15:27:16 发布

Charliefive

最新推荐文章于 2024-11-24 15:27:16 发布

阅读量2.1k

点赞数 4

分类专栏： python 文章标签： python pandas

本文链接：https://blog.csdn.net/weixin_43886282/article/details/127653764

版权

python 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了使用Pandas库中的stack和unstack函数进行数据重塑的方法。通过实例演示了如何将表格结构的数据转换为花括号结构，并能逆向操作。同时展示了如何通过reset_index和set_index方法灵活地调整数据的索引。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、介绍

在用pandas进行数据重排时，经常用到stack和unstack两个函数。

stack的意思是堆叠，堆积，unstack即“不要堆叠”，

我对两个函数是这样理解和区分的。

常见的数据的层次化结构有两种，一种是“花括号”（下图左），一种是表格（下图右），即下面这样的这两种形式：

在这里插入图片描述
花括号结构只有“列方向”上的索引（类似于层次化的Series），结构更加偏向于堆叠（Series-stack，方便记忆）。

表格在行列方向上均有索引（类似于DataFrame）。

stack函数会将数据从”表格结构“变成”花括号结构“，即将其行索引变成列索引，

unstack函数将数据从”花括号结构“变成”表格结构“，即要将其中一层的列索引变成行索引。

二、代码

2.1准备数据

data = pd.DataFrame(np.arange(6).reshape((2, 3)), index=['street1', 'street2'], columns=['one', 'two', 'three'])
print(data)

         one  two  three
street1    0    1      2
street2    3    4      5

2.2转换成“花括号结构”

data2 = data.stack()
print(data2)

street1  one      0
         two      1
         three    2
street2  one      3
         two      4
         three    5
dtype: int32

2.3重置索引，将索引列变成正常列

data3=data2.reset_index()
print(data3)

在这里插入图片描述

2.4可以将正常列变成索引列

data4=data3.set_index(['level_0','level_1'])
print(data4)

在这里插入图片描述

2.5其从“花括号结构”变成“表格结构”

data5=data4.unstack()
data5

在这里插入图片描述

2.6可以重置索引，将索引列变成正常列

data6=data5.reset_index()
data6

在这里插入图片描述