python-2.7 - 如何在 pandas DataFrame 中选择性地乘法或添加列？

Question

我有一个如下所示的数据框:

Sample    P1    P2    P3
A         0.25  0.5   0.25
B         0.25  0     0.25
C         0.25  0.25  0

我希望能够有选择地仅将 P 列相乘(如果它们不等于零)，并输出第五列相乘值。它应该如下所示:

Sample    P1    P2    P3    Multiplied
A         0.25  0.5   0.25  0.03125
B         0.25  0     0.25  0.0625
C         0.25  0.25  0     0.0625

我不太确定如何为此编写代码。对此有什么建议吗？

Answer 1

只是为了抛出一个替代方案(您可以使用 prod DataFrame 方法):

In [11]: df['multiplied'] = df[df != 0].prod(axis=1)

In [12]: df
Out[12]: 
          P1    P2    P3  multiplied
Sample                              
A       0.25  0.50  0.25     0.03125
B       0.25  0.00  0.25     0.06250
C       0.25  0.25  0.00     0.06250