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

Question

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

我以为我能做到:

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它替换它没有警告.

Answer 1

你需要设置invalid而不是divide:

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