[numpy除零自动纠正] numpy np div 0 #numpy

numpy_div0.py

# 自动处理除零错误
# numpy >= 1.7
a = np.array([-1, 0, 1, 2, 3], dtype=float)
b = np.array([ 0, 0, 0, 2, 2], dtype=float)

# If you don't pass `out` the indices where (b == 0) will be uninitialized!
c = np.divide(a, b, out=np.zeros_like(a), where=b!=0)
print(c)
# [ 0.   0.   0.   1.   1.5]

# 如果需要除0输出正无穷, 可以 out=np.full_like(a, fill_value=np.inf)