我怎样才能在一组熊猫中区分行呢？

Question

我有这样的数据：

import pandas as pd

df = pd.DataFrame({'group': [1, 1, 1, 2, 2, 3, 3, 3, 3], 'time': [12, 44, 55, 2, 7, 100, 105, 106, 200]})

#    group  time
# 0      1    12
# 1      1    44
# 2      1    55
# 3      2     2
# 4      2     7
# 5      3   100
# 6      3   105
# 7      3   106
# 8      3   200

我想得到这样的东西：

#    group  time diff_to_next_step
# 0      1    12  32     (44-12)
# 1      1    44  11     (55-44)
# 2      1    55  None
# 3      2     2  5      (7-2)
# 4      2     7  None
# 5      3   100  5      (105 - 100)
# 6      3   105  1      (106 - 105)
# 7      3   106  94     (200 - 106)
# 8      3   200  None

我认为这在某种程度上应该适用于组/堆栈/轮班，但我目前还不能让它发挥作用。有人能帮帮我吗？

Answer 1

尝试使用shift

df['new'] = df.groupby('group')['time'].shift(-1) - df.time

Answer 2

使用diff + shift

import pandas as pd

df = pd.DataFrame({'group': [1, 1, 1, 2, 2, 3, 3, 3, 3], 'time': [12, 44, 55, 2, 7, 100, 105, 106, 200]})

df["diff"] = df.groupby("group").transform("diff").shift(-1)
print(df)

输出

   group  time  diff
0      1    12  32.0
1      1    44  11.0
2      1    55   NaN
3      2     2   5.0
4      2     7   NaN
5      3   100   5.0
6      3   105   1.0
7      3   106  94.0
8      3   200   NaN