Pandas-以如果值大于0为条件，将一列除以另一列？

Question

我有一个包含日期、项目和 2 个值的 pandas 数据框。我要做的就是输出另一列，如果 B 列大于 0，则输出 A 列 / B 列的乘积，如果 B 列等于 0，则输出 0。

   date     item   A   B        C       
 1/1/2017   a      0   3             0  
 1/1/2017   b      2   0             0  
 1/1/2017   c      5   2           2.5  
 1/1/2017   d      4   1             4  
 1/1/2017   e      3   3             1  
 1/1/2017   f      0   4             0  
 1/2/2017   a      3   3             1  
 1/2/2017   b      2   2             1  
 1/2/2017   c      3   9   0.333333333  
 1/2/2017   d      4   0             0  
 1/2/2017   e      5   3   1.666666667  
 1/2/2017   f      3   0             0  

这是我编写的代码，但内核一直在死机（请记住，这只是一个示例表，我有大约 30,000 行，所以没什么太疯狂的）

df['C'] = df.loc[df['B'] > 0, 'A'] / df['B'])

知道发生了什么吗？是否有东西无限运行导致它崩溃？谢谢您的帮助。

Answer 1

您可以使用 np.where 得到它

df['C'] = np.round(np.where(df['B'] > 0, df['A']/df['B'], 0), 1)

或者如果你想使用loc

df.loc[df['B'] > 0, 'C'] = df['A']/df['B']

然后fillna(0)

Answer 2

选项 1
您使用pd.Series.mask 隐藏零，然后使用fillna 清空单元格。

v = (df.A / df.B.mask(df.B == 0)).fillna(0)
v

0     0.000000
1     0.000000
2     2.500000
3     4.000000
4     1.000000
5     0.000000
6     1.000000
7     1.000000
8     0.333333
9     0.000000
10    1.666667
11    0.000000
dtype: float64

df['C'] = v

或者，将这些零替换为np.inf，因为x / inf = 0。

df['C'] = (df.A / df.B.mask(df.B == 0, np.inf))

---

选项 2
直接替换为df.replace

df.A / df.B.replace(0, np.inf)

0     0.000000
1     0.000000
2     2.500000
3     4.000000
4     1.000000
5     0.000000
6     1.000000
7     1.000000
8     0.333333
9     0.000000
10    1.666667
11    0.000000
dtype: float64

请记住，如果您想要混合整数和浮点数作为结果，您可以进行 astype 转换：

df.A.div(df.B.replace(0, np.inf)).astype(object)

0            0
1            0
2          2.5
3            4
4            1
5            0
6            1
7            1
8     0.333333
9            0
10     1.66667
11           0
dtype: object