数据帧中按组键的 Python 值差异

Question

我有一个数据框

name   value
A       2
A       4
A       5
A       7
A       8
B       3
B       4
B       8
C       1
C       3
C       5 

我想根据每个名字获得值(value)差异像这样

name   value   dif
A       2      0
A       4      2
A       5      1
A       7      2
A       8      1
B       3      0
B       4      1
B       8      4
C       1      0
C       3      2
C       5      2

谁能告诉我最简单的方法？

Answer 1

您可以使用 GroupBy.diff计算每个分组对象的连续行之间的差异。可选地，用 0 填充缺失值(每组中的第一行)并将它们最终转换为整数。

df['dif'] = df.groupby('name')['value'].diff().fillna(0).astype(int)
df