python - 根据条件创建列

Question

我有这个数据集:

        Players       Material_bough
1         X1                Wood
2         X2                Iron
3         X1                Stone

我想创建一个新列来显示玩家购买的 Material 总和。

        Players       Material_bough         Material_total
1         X1                Wood                  2
2         X2                Iron                  1
3         X1                Stone                 2

木头 + 石头 = 2 种 Material 。

我认为我应该使用 groupby 但不知道如何。

Answer 1

我认为这与计数 Players 相同，无需按 transform 聚合:

df['Material_total'] = df.groupby('Players')['Material_bough'].transform('size')

或者:

df['Material_total'] = df.groupby('Players')['Material_bough'].transform('count')