python - 如何有条件地更新 np.array ？

Question

我有两个 numpy 数组:

Out[23]: a = np.array([3, 7, 8, 9, 2], dtype=float)
Out[24]: b = np.array([5, 6, 7, 10, 3], dtype=float)

我需要比较 a 和 b，并更新数组 b 使其元素包含较高的值或 np.nan

示例代码:

for i in range(len(b)):
    b[i] = b[i] if b[i] > a[i] else np.nan

结果:

b
Out[33]: array([  5.,  nan,  nan,  10.,   3.])

有没有办法在不使用 for 循环的情况下做到这一点？

提前致谢。

Answer 1

您可以使用np.where

>>> np.where(b > a, b, np.nan)
array([  5.,  nan,  nan,  10.,   3.])