2017-03-05 91 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 –