python - Pandas:在组中将值向下移动一行

Question

我有一个 Pandas 数据框，我想创建一个新列，其值是另一列的值，向下移动一行。最后一行应该显示 NaN。

问题是我想按组执行此操作，每组的最后一行显示 NaN。不要让组的最后一行“窃取”数据帧中恰好相邻的组中的值。

我尝试的实现被严重破坏了，所以我显然误解了一些基本的东西。

df['B_shifted'] = df.groupby(['A'])['B'].transform(lambda x:x.values[1:])

Answer 1

新版本的 pandas 现在可以对组执行 shift:

df['B_shifted'] = df.groupby(['A'])['B'].shift(1)

请注意，当 向下 移动时，它是具有 NaN 的 第一 行。