python - 替换 numpy 数组中的负值

Question

有没有一种简单的方法可以用 0 替换数组中的所有负值？

我对如何使用 NumPy 数组进行了完整的了解。

例如

a = array([1, 2, 3, -4, 5])

我要回去

[1, 2, 3, 0, 5]

a < 0给出:

[False, False, False, True, False]

这就是我卡住的地方——如何使用这个数组来修改原始数组。

Answer 1

你已经成功了一半。试试:

In [4]: a[a < 0] = 0

In [5]: a
Out[5]: array([1, 2, 3, 0, 5])