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

一、介绍

在用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