python pandas df：根据另一列中该行的值添加到一列

Question

我的df看起来像：

    a   b   c   d
0   0   0   0   0
1   0   0   0   0
2   1   292 0   0
3   0   500 1   406
4   1   335 0   0

我想找到列b的总和，其中a = 1表示该行。所以在我的例子中，我希望添加第2行和第4行（只是第b列），而不是第3行。如果它有任何区别，那么只有0和1。谢谢你的帮助！

Answer 1

您需要使用.loc

>>> df.loc[df.a==1, 'b'].sum()
627

您可以查看文档here以进行索引和选择数据。