python - numpy.where : how to delay evaluating parameters?

Question

我正在使用numpy.where，我想知道是否有一种简单的方法可以避免调用未使用的参数。示例:

import numpy as np
z = np.array([-2, -1, 0, 1, -2])
np.where(z!=0, 1/z, 1)

返回:

array([-0.5, -1. ,  1. ,  1. , -0.5])

但我收到除以零的警告，因为当 z 为 0 时，代码仍然会计算 1/z，即使它不使用它。

Answer 1

您还可以关闭警告并在完成后将其重新打开使用上下文管理器errstate :

with np.errstate(divide='ignore'):
    np.where(z!=0, 1/z, 1)