python - 如何通过id对数据进行分组，并找出相邻数据的差异？

Question

我有以下数据

id    starting_point      ending_point
A        2525                 6565
B        5656                 8989
A        1234                 5656
A        4562                 6245
B        6496                 9999
B        1122                 2211

对于以上数据:

df['id'] = ['A','B','A','A','B', 'B']
df['starting_point'] =['2525','5656','1234','4562','6496','1122']
df['ending_point'] = ['6565','8989','5656','6245','9999','9999']

我想编写一个 python 代码，按它们的 id(即 A、B、...)对它们进行分组，并找出 A 的第一个和第二个终点之和与A. 在这种情况下 [(6565+5656) - (5656+6245)].

Answer 1

IIUC 你可以 scip 第二个终点，如 (6565+5656) - (5656+6245) == 6565 - 6245:

In [15]: df.groupby('id')['ending_point'].apply(lambda x: x - x.shift(-2))
Out[15]:
0     320.0
1    6778.0
2       NaN
3       NaN
4       NaN
5       NaN
Name: ending_point, dtype: float64