pivot_table的应用

import pandas as pd

df = pd.DataFrame({‘timestamp’: [‘11:00’, ‘11:00’, ‘12:00’, ‘12:00’],
‘type’: [‘a’, ‘b’, ‘a’, ‘b’],
‘v’: [10, 2, 20, 8]
})
timestamp type v
0 11:00 a 10
1 11:00 b 2
2 12:00 a 20
3 12:00 b 8

想换成：
timestamp a b
0 11:00 10 2
1 12:00 20 8

pivot = pd.pivot_table(data=df, columns=df[‘type’], index=df[‘timestamp’])
df2 = pivot.reset_index()
print(df2, ‘\n’)

得到：
timestamp v
type a b
0 11:00 10 2
1 12:00 20 8

得到：
然而这样得到的数据表表头部分其实是多重索引的，也就是这样：

MultiIndex([(‘v’, ‘a’),
(‘v’, ‘b’)],
names=[None, ‘type’])

这样通常是不能够直接进行后面数据处理操作的。因此这里建议将此重新保存为dataFrame，用到的函数可以是
pivot.reset_index()

然后：
columns = df[“type”].unique().tolist() #获取列的不重复值，并保存为list
columns = [‘timestamp’] + columns # list的合并，直接 +
df2.columns = columns #定义dataframe的列名
print(df2)

最终：
timestamp a b
0 11:00 10 2
1 12:00 20 8