计算基于另一个col. 的列的和

Question

我的df是这样的:

value    type
12       x
34       z
54       x
14       y

我想创建一个新的列df.sum，我想对值col求和，但只在type == x。其余行应该为空。例如，输出应该是这样的:

value    type    sum
12       x       86
34       z
54       x       86
14       y

Answer 1

如果您想处理单一类型(仅x):

mask = df['type'].eq('x')
df.loc[mask, 'sum'] = df.loc[mask, 'value'].sum()

如果你可能需要处理几个:

types = ['x'] # add others, e.g.: types = ['x', 'y']
df.loc[df['type'].isin(types), 'sum'] = (df.groupby('type')['value']
.transform('sum')
)

输出:

value type   sum
0     12    x  66.0
1     34    z   NaN
2     54    x  66.0
3     14    y   NaN

Answer 2

是啊，它看起来很奇怪，但仍然可以工作:

types = ['2','x']  # your keys to sum
df = df.merge(df.query('type in @types').
groupby('type', as_index=False).
agg(sum), 
how='left', on='type', suffixes=(None,'_sum'))
'''
value type  value_sum
0     12    x       66.0
1     34    z        NaN
2     54    x       66.0
3     14    y        NaN