python - 使用 groupby 创建具有最大值的新列

Question

从以下数据框中，我尝试添加一个新列，条件是为每个 id 检查最大值。然后将每个 id 的每一行的最大值放在新列中。

df

id  value
1   0
1   0
1   0
2   0
2   1
3   1
3   1

预期结果:

id  value new_column
1   0     0
1   0     0
1   0     0
2   0     1
2   1     1
3   1     1
3   1     1

我试过:

df['new_column'] = df.groupby(['id'])['value'].idxmax()

或者:

df['new_column'] = df.groupby(['id'])['value'].max()

但是这些都没有给出预期的结果。

Answer 1

你需要为此使用transform:

df['new_column'] = df.groupby(['id'])['value'].transform('max')

这更简洁地复制了以下内容:

df['new_column'] = df['id'].map(df.groupby(['id'])['value'].max())

请记住，groupby 操作的结果是一个索引设置为石斑鱼列的系列。

由于索引在您的原始数据框和groupby对象之间没有对齐，分配不会自动发生。