如何避免" RuntimeWarning:在div"中遇到无效值在NumPy?

时间:2014-09-07 11:45:09

标签: python numpy

我试图避免在NumPy中发出警告RuntimeWarning: invalid value encountered in divide

我以为我能做到:

import numpy as np

A=np.array([0.0])
print A.dtype
with np.errstate(divide='ignore'):
    B=A/A
print B

但这给出了:

float64
./t.py:9: RuntimeWarning: invalid value encountered in divide
  B=A/A
[ nan]

如果我将B=A/A替换为np.float64(1.0) / 0.0,则不会发出警告。

1 个答案:

答案 0 :(得分:25)

您需要设置invalid而不是divide

with np.errstate(invalid='ignore'):
                 ^^^^^^^