如何通过比较具有共同变量值的不同行来创建新变量？

import pandas as pd

# 创建示例数据框
data = {
    'ID': [1, 2, 3, 4, 5],
    'Category': ['A', 'A', 'B', 'B', 'C'],
    'Value': [10, 15, 20, 25, 30]
}
df = pd.DataFrame(data)

# 按 Category 分组，并计算相邻行的 Value 差值
df['Diff_Value'] = df.groupby('Category')['Value'].diff()

print(df)

   ID Category  Value  Diff_Value
0   1        A     10         NaN
1   2        A     15          5.0
2   3        B     20         NaN
3   4        B     25          5.0
4   5        C     30         NaN

# 填充缺失值
df['Value'].fillna(method='ffill', inplace=True)

# 重新计算差值
df['Diff_Value'] = df.groupby('Category')['Value'].diff()

def custom_agg(group):
    group['Custom_Value'] = group['Value'].max() - group['Value'].min()
    return group

df = df.groupby('Category').apply(custom_agg).reset_index(drop=True)