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

时间:2016-12-07 07:16:39

标签: python pandas dataframe

我有一个DataFrame

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

我希望根据每个名字获得价值差异 像这样

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

有人能告诉我最简单的方法吗?

1 个答案:

答案 0 :(得分:5)

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

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

enter image description here