熊猫据帧 - 沁列WRT到值的另一列

Question

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

data = {"doc1" : {'a': 2 , 'b': 1,'c':3}, "doc2" : {'a': 1 , 'b': 1,'c':3}, "doc3" : {'a': 1 , 'b': 1,'c':3}} 

我把它转换成一个数据帧： -

df = pd.DataFrame.from_dict(data,orient='index') 

数据框的样子这一点： -

a c b doc1 2 3 1 doc2 1 3 1 doc3 1 3 1

现在我要总结列BW中的所有值这里列的值是1

所以我想的值将是2

是否有一个简单的方法来做到这一点，而不是通过这两个列迭代？ 我查了其他帖子，发现this： -

这使得使用.loc函数。 df.loc[df['a'] == 1, 'b'].sum()

但由于某种原因，我似乎无法使它与我的数据框一起工作。

请让我知道。

感谢。

Answer 1

你非常接近。见下文。

>>> df[df['a'] == 1]['b'].sum() 
2 

代替使用.loc的，尽量只滤波数据帧的第一（df[df['a'] == 1]），则选择列'b'，然后求和。

编辑：我会留下这里供将来参考，虽然取决于你使用的熊猫版本，你的解决方案应该工作（谢谢，@maxymoo）。我正在运行0.18.1，并且这两种方法都有效。