转置pandas数据帧的一部分

Question

我有以下数据框：

>>> df
Out[15]: 
      group   type  amount  number
0   group_A    buy     100     123
1   group_A   view       0     111
2   group_B   view       0     222
3   group_A   view       0     222 

我想把数据旋转，以便最终得到：

              type  group_A   group_B
0    amount    buy      100         0
1    number    buy        0       123
2    number   view      333       222

我该如何实现这个？

- greenafrican

Answer 1

使用：

df=pd.DataFrame([['group_A','buy',100,123],['group_A','view',0,111],['group_B','view',0,222],['group_A','view',0,222]],columns=['group','type','amount','number'])

首先将索引相加并确定方向：

>>> df = df.groupby(['type','group']).sum().transpose().stack(0).reset_index()
>>> df
group level_0  type  group_A  group_B
0      amount   buy      100      NaN
1      amount  view        0        0
2      number   buy      123      NaN
3      number  view      333      222

删除所有值都为零的行：

df = df[~((df['group_A']==0) | (df['group_B']==0))]

填充缺失值：

>>> df.fillna(0)
group level_0  type  group_A  group_B
0      amount   buy      100        0
2      number   buy      123        0
3      number  view      333      222

这里有一些地方我猜测了一下，但它应该能为您提供一个开始。

- Daniel

---

谢谢。我之前一直在尝试使用多重索引，但这种方法简单得多。 - greenafrican