汇总 Pandas DataFrame 中因子水平的差异？

Question

给定df1：

  A B C
0 a 7 x
1 b 3 x
2 a 5 y
3 b 4 y
4 a 5 z
5 b 3 z

如何获取df2 where 对于df1的C中的每个值，新的col D在col A==a和col A==b的col B中的df1值之间存在差异：

  C D
0 x 4
1 y 1
2 z 2

Answer 1

我会使用 pivot 表：

df = df1.pivot_table(columns = ['A'],values = 'B', index = 'C')
df2 = pd.DataFrame({'D': df['a'] - df['b']})

@YOBEN_S 给出的答案中的风险是，如果b对于给定的C值出现在a之前，它将失败