python - 将 Numpy 数组除以标量，其中 Array-Element 低于某个值

Question

我有一个像

[[1, 2, 3],
 [4, 5, 6],
 [7, 8, 9]]

现在我想将小于 5 的每个元素除以 2。结果应该是

[[0.5, 1, 1.5],
 [2, 5, 6],
 [7, 8, 9]]

我怎样才能做到这一点？

标签： pythonnumpynumpy-ndarray

Answer 1

使用numpy.where：

np.where(arr<5, arr/2, arr)

输出：

array([[0.5, 1. , 1.5],
       [2. , 5. , 6. ],
       [7. , 8. , 9. ]])