Python 2.7 如何在pandas中向dataframe列的一部分添加数字？

我有一个包含两列a和B的数据帧

A        B
1        0
2        0
3        1
4        2
5        0
6        3

我想做的是将A列与B列相加，但只与B列对应的非零值相加，并将结果放在B列上

A        B
1        0
2        0
3        4
4        6
5        0
6        9

---

提前感谢您的帮助和建议。

使用带有布尔掩码的

.loc

：

In [49]:
df.loc[df['B'] != 0, 'B'] = df['A'] + df['B']
df

Out[49]:
   A  B
0  1  0
1  2  0
2  3  4
3  4  6
4  5  0
5  6  9