在分组的pandas数据帧中创建两个移位列

import pandas as pd

# 创建示例数据帧
df = pd.DataFrame({'Group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
                   'Value': [1, 2, 3, 4, 5, 6, 7]})

# 定义移位函数
def create_shifted_columns(group):
    group['Shifted_1'] = group['Value'].shift(1)
    group['Shifted_2'] = group['Value'].shift(2)
    return group

# 对数据帧进行分组并应用移位函数
df = df.groupby('Group').apply(create_shifted_columns)

# 打印结果
print(df)

  Group  Value  Shifted_1  Shifted_2
0     A      1        NaN        NaN
1     A      2        1.0        NaN
2     B      3        NaN        NaN
3     B      4        3.0        NaN
4     B      5        4.0        3.0
5     C      6        NaN        NaN
6     C      7        6.0        NaN