使用分区计算Pandas中条目之间的增量

Question

我在Pandas中使用了Pandas，我想使用一个分区计算相邻行之间的增量。

例如，这是我按A和B排序后的初始集：

    A   B    
1   12  40
2   12  50
3   12  65
4   23  30
5   23  45
6   23  60

我想要计算由A划分的相邻B值之间的增量，如果我们将C定义为结果，最后的表应该如下所示：

    A   B   C   
1   12  40  NaN
2   12  50  10
3   12  65  15
4   23  30  NaN
5   23  45  15
6   23  75  30

NaN的原因是我们不能计算每个分区中最小数目的增量。

Answer 1

您可以按A列分组，并接受以下差异：

df['C'] = df.groupby('A')['B'].diff()

df
Out: 
    A   B     C
1  12  40   NaN
2  12  50  10.0
3  12  65  15.0
4  23  30   NaN
5  23  45  15.0
6  23  60  15.0