pandas - 使 Pandas 在除以零而不是 inf 时提高

Question

我想让 Pandas 在除以零时引发异常，如下所示:

d = {'col1': [2., 0.], 'col2': [4., 0.]}
df = pd.DataFrame(data=d)
2/df

而不是当前的结果:

0    1.000000
1         inf
Name: col1, dtype: float64

任何建议如何实现这一目标？

我知道 numpy 我可以 np.seterr(divide='raise')但 Pandas 确实忽略了这一点。

非常感谢

Answer 1

这远非理想，但一种潜在的选择是将数据帧的元素解释为 Python 对象，而不是更优化的 numpy或 pandas它通常使用的数据类型:

In [37]: d = {'col1': [2., 0.], 'col2': [4., 0.]}
    ...: df = pd.DataFrame(data=d)
    ...: 2/df

Out[37]:
   col1  col2
0   1.0   0.5
1   inf   inf

In [38]: 2 / df.astype('O')
---------------------------------------------------------------------------
ZeroDivisionError: float division by zero