为列pandas dataframe中的第一个/最大值更改创建dummy by group

import pandas as pd

# 加载数据到dataframe
df = pd.read_csv('your_data.csv')

# 按照某列进行分组
grouped = df.groupby('group_column')

# 定义自定义函数来更改第一个/最大值
def change_first_max_value(group):
    # 找到第一个/最大值的索引
    index = group.idxmax()  # 或者使用idxmin()找到最小值的索引

    # 更改第一个/最大值为1，其他值为0
    group.loc[group.index == index] = 1
    group.loc[group.index != index] = 0

    return group

# 应用自定义函数到每个分组
df['dummy'] = grouped['value_column'].apply(change_first_max_value)

print(df)