使用GroupBy计算熊猫数据框中的自定义速率函数

import pandas as pd

# 加载数据框
df = pd.DataFrame(...)

def custom_rate_func(group):
    # 计算每个组的平均值
    avg = group.mean()
    
    # 计算每个组的差异百分比
    diff_percentage = (group - group.shift(1)) / group.shift(1) * 100
    
    return diff_percentage

# 按照指定列进行分组
grouped = df.groupby('column_name')

# 应用自定义速率函数
result = grouped['column_name'].apply(custom_rate_func)

# 将结果保存到新的列中
df['custom_rate'] = result