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

Question

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

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

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

我想计算相邻 B 值之间的增量，按 A 划分。如果我们将 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