python - 如何根据条件计算熊猫行的权重

Question

我有一个看起来像这样的数据框

pd.DataFrame({'A': ['C1', 'C2', 'C3', 'C4', 'C5'],
  ...:                    'B': [6, 1, 7, 4, 3],
  ...:                    'C': [True, True, False, False, True]})
Out[6]: 
    A  B      C
0  C1  6   True
1  C2  1   True
2  C3  7  False
3  C4  4  False
4  C5  3   True

对于 C 列 = True 的所有行，我想根据 B 列计算权重。结果数据框应如下所示：

    A  B      C   weight
0  C1  6   True   0.6
1  C2  1   True   0.1
2  C3  7  False   0.0
3  C4  4  False   0.0
4  C5  3   True   0.3

适用的逻辑：

B 列的总和，其中 C = True = 10 (6+1+3)

所以对于第 0 行（C1）“权重”= 6 / B 的总和，其中 C = True

我怎样才能用 1 行代码实现这一点？

标签： pythonpandas

Answer 1

你可以做

df['Weight'] = df['B']*df['C']/sum(df['B']*df['C'])
df
Out[136]: 
    A  B      C  Weight
0  C1  6   True     0.6
1  C2  1   True     0.1
2  C3  7  False     0.0
3  C4  4  False     0.0
4  C5  3   True     0.3