使用Python numpy where条件将整数更改为低于特定值

Question

这是我的数组：

import numpy as np
a = np.array([0, 5.0, 0, 5.0, 5.0])

是否可以通过numpy.where某种方式x为所有a小于的值增加一个值l？

所以像这样：

a = a[np.where(a < 5).add(2.5)]

应该返回：

array([2.5, 5.0, 2.5, 5.0, 5.0])

Answer 1

a = np.array([0., 5., 0., 5., 5.])
a[np.where(a < 5)] += 2.5

万一您真的想使用where还是

a[a < 5] += 2.5

我通常将其用于此类操作。