熊猫与Groupby换位

时间:2020-02-05 13:11:22

标签: python python-3.x pandas python-2.7

我有一个简单的熊猫数据框,该数据框由三列以及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,但努力获取任何可行的方法。

0 个答案:

没有答案