如何获取基于pandas分组的变量的滞后值？

import pandas as pd

# 创建示例数据
data = {'group': ['A', 'A', 'A', 'B', 'B', 'B'],
        'value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)

# 按照分组进行排序
df.sort_values(by=['group'], inplace=True)

# 定义一个函数来获取滞后值
def get_lag_value(x):
    x['lag_value'] = x['value'].shift(1)
    return x

# 应用函数到每个分组
df = df.groupby('group').apply(get_lag_value)

print(df)

  group  value  lag_value
0     A      1        NaN
1     A      2        1.0
2     A      3        2.0
3     B      4        NaN
4     B      5        4.0
5     B      6        5.0