Python Pivot-Table中的列差异

Question

我有一个pandas pivot_table，可以在两行中聚合2个数据集。我想添加另一列，即两行现有列中的聚合值之间的差异。有没有办法直接在pivot_table（）调用中实现它？我知道返回的数据帧是一个数据帧，所以我可以通过其他方式计算它，但只是好奇是否有更有效的方法。

我的数据的简单示例：

  Set     Type   Val
  S1       A     1
  S1       B     2
  S1       B     3
  S2       A     4
  S2       B     5
  S2       C     6

使用以下代码，其中数据是我的df

piv=pivot_table(data,'Val',rows='Type',cols='Set',aggfunc=sum,fill_value=0.0)

我得到以下

    S1  S2
A   1   4
B   5   5
C   0   6

我希望输出为

    S1  S2 Diff
A   1   4   3
B   5   5   0
C   0   6   6

或只是

   Diff
A   3
B   0
C   6

Answer 1

易。数据帧（以及一般的矩阵）使得一次操作多个元素变得容易。

定义您要应用的功能。

>>> def abs_diff(x, y):
>>>     return abs(x - y)

然后，应用它。

>>> df['Diff'] = abs_diff(df['S1'], df['S2'])

>>> df

   S1  S2  Diff
A   1   4     3
B   5   5     0
C   0   6     6

当然，如果您只想呈现特定列： -

>>> df['Diff']

A    3
B    0
C    6
Name: Diff

（>>>当然是python shell提示符）