擅长:python、mysql、java
<p>IIUC公司</p>
<pre><code>reduced['prob_a'] = data.loc[(data.a == 1) & (data.total == 1),'rate'].values[0]
reduced['prob_b'] = data.loc[(data.b == 1) & (data.total == 1),'rate'].values[0]
reduced['prob_c'] = data.loc[(data.c == 1) & (data.total == 1),'rate'].values[0]
reduced[['prob_a','prob_b','prob_c']]=reduced[['prob_a','prob_b','prob_c']].mul(reduced[['a','b','c']].eq(1).values)
reduced
Out[698]:
a b c rate total prob_a prob_b prob_c
4 1 1 0 0.24 2 0.1 0.11 0.00
5 1 0 1 0.27 2 0.1 0.00 0.12
</code></pre>