将熊猫数据帧从宽到长

Question

我试图将pandas数据帧从宽到长，但我找不到一个好方法。有没有建议通过熊猫来实现这一点？

data = {'colour':["red", "blue", "yellow", "red", "yellow"],
        'apple':[1, 2, 4, 5, 6], 
        'organge': [3, 4, 5, 7, 8],
        'watermelon': [7, 8, 9, 1, 0]}
df = pd.DataFrame(data)
df

   colour  apple  organge  watermelon
0     red      1        3           7
1    blue      2        4           8
2  yellow      4        5           9
3     red      5        7           1
4  yellow      6        8           0

预期结果：

        fruit  red  blue  yellow
0       apple    6     2      10
1     organge   10     4      13
2  watermelon    8     8       9

Answer 1

因此，您可以使用< code>T执行< code>groupby

out = df.groupby('colour').sum().T
Out[45]: 
colour      blue  red  yellow
apple          2    6      10
organge        4   10      13
watermelon     8    8       9

或

out = df.set_index('colour').T.sum(level=0,axis=1)
Out[48]: 
colour      red  blue  yellow
apple         6     2      10
organge      10     4      13
watermelon    8     8       9