使用numpy.where过滤0元素避免除零警告失效，原因何在？

你遇到的情况是：使用numpy.where()尝试过滤数组中等于0的元素以避免除零错误，虽然最终返回结果正确，但会触发RuntimeWarning；而用简单for循环分支处理时完全没有警告，即使显式指定数组dtype也无法消除np.where()的警告。

测试代码如下：

import numpy as np

# test array
x = np.array([1e-20, 1, 10, 0], dtype=np.float64)

# divide by x except where x == 0 
np.where(x == 0, 1/np.pi, np.sin(x)/(np.pi*x))
# returns correct values but with RuntimeWarning 

# for loop works with no warnings appearing
for x_i in x:
    if x_i == 0:
        print(1/np.pi)
    else:
        print(np.sin(x_i)/(np.pi*x_i))

核心原因：两者执行逻辑完全不同 #

• np.where()的执行逻辑：它会先完整计算所有传入的三个参数表达式，再根据条件数组选择对应位置的结果。也就是说，不管x == 0的条件是否满足，np.sin(x)/(np.pi*x)这个表达式会对整个数组x的所有元素执行计算——包括那个值为0的元素，这就必然触发除零警告，哪怕最终结果里不会用到这个错误计算的结果。
• for循环的执行逻辑：是逐个元素判断分支执行。当遇到值为0的元素时，只会执行1/np.pi的计算，根本不会进入else分支去执行np.sin(x_i)/(np.pi*x_i)，自然不会触发除零警告。

可选解决办法 #

如果要使用numpy的向量化操作同时避免警告，可以用两种方式：

1. 临时抑制警告：

import numpy as np

x = np.array([1e-20, 1, 10, 0], dtype=np.float64)
with np.errstate(divide='ignore', invalid='ignore'):
    result = np.where(x == 0, 1/np.pi, np.sin(x)/(np.pi*x))

2. 先处理数组，替换0值后再计算：

import numpy as np

x = np.array([1e-20, 1, 10, 0], dtype=np.float64)
x_safe = np.copy(x)
x_safe[x_safe == 0] = 1  # 临时替换0为1，避免除零
result = np.sin(x_safe)/(np.pi*x_safe)
result[x == 0] = 1/np.pi

内容的提问来源于stack exchange，提问作者yyyLLL