2017-03-05 94 views
0

我在寻找是否有直接的方法来计算b的元素总和时a=1连续两列熊猫python3

a 

    a b  c d 
0 4 2 NaN 7 
1 1 2 45.0 74 
2 1 4 7.0 8 

所以这种情况下,它必须是6 我可以用我自己的代码很容易做,但我有一个大的数据集,所以我想知道做到这一点的最好办法。 谢谢。

import pandas as pd 
a=pd.DataFrame([[4,2,None,7],[1,2,45,74],[1,4,7,8]],columns=(['a','b','c','d'])) 

回答

1

是否

a.loc[a.a == 1].b.sum() 

工作?

+0

thak you,it works –