python - 以简单的方式操作数组值

Question

从上一篇文章中，我学习了如何以非常快速且简单的方式更改 2D 数组中的所有值:

例如将所有小于 255 的值设置为 x:

arr[arr < 255] = x

现在我想做类似的事情，但不仅仅是将值设置为 x。

当该数组中的值低于阈值时，我想以特定方式对其进行操作。

arr[abs(arr) < threshold] = arr(*where condition is true*) - threshold*x

arr - theshold*x 不起作用，因为 arr 是完整的数组。

我认为 np.where 是一个循环需要解决这个问题，这不像以前那么容易。还有更好的办法吗？

Answer 1

扩展您已经完成的工作应该很容易:

x = random.rand(100)
threshold = 0.2
indices = abs(x) < threshold
x[indices] = x[indices] - threshold*x[indices]