向上移动列 basis groupby

Question

现有数据框：

Id       event      time_spent_in_sec
A         in               0
A         step_1           2.2
A         step_2           3
A         done             3
B          in              0
B         step_1           5
B         step_2           8
B         step_3           15
B         done             7

预期数据框：

Id       event      time_spent_in_sec
A         in               2.2
A         step_1           3
A         step_2           3
A         done             0
B          in              5
B         step_1           8
B         step_2           15
B         step_3           7
B         done             0

我希望移动列中的值time_spent_in_sec并用 0 填充每个唯一 ID 的最后一行。

我尝试使用 shift(1) 但坚持用 0 填充最后一行

Answer 1

您可以使用.fillna() 将其填充为原始的第一个数字：

df.time_spent_in_sec.shift(-1).fillna(df.time_spent_in_sec[0])

或者：

df.time_spent_in_sec.shift(-1, fill_value = df.time_spent_in_sec[0])