我有一个简单的熊猫数据框,该数据框由三列以及id,stage和value组成。每个ID在两个阶段的每一阶段都有1行。我想转置这些数据以准备将其用于堆积面积图,但遇到了一些麻烦。
初始数据帧如下:
id stage value
0 1 second 3
1 1 first 4
2 2 first 5
3 2 second 2
,其生成方式如下:
import pandas as pd
df = pd.DataFrame(data=[['1', 'second', 3],
['1', 'first', 4],
['2', 'first', 5],
['2', 'second', 2]],
columns=['id', 'stage', 'value'])
所需的输出数据帧如下:
id first second
0 1 4 3
1 2 5 2
我知道这将涉及移调和groupby,但努力获取任何可行的方法。