如何在numpy数组上执行条件数组算术？

小编典典

python

我试图更好地掌握numpy数组，所以我有一个关于它们的示例问题：

假设我有一个名为a的numpy数组。我想对进行操作，使其中的所有小于0的值递增，其余的保持不变。例如，如果我有：

a = np.array([1,2,3,-1,-2,-3])

我想返回：

([1,2,3,0,-1,-2])

最紧凑的语法是什么？

谢谢！

阅读 124

2020-12-20

共1个答案

小编典典

In [45]: a = np.array([1,2,3,-1,-2,-3])

In [46]: a[a<0]+=1

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

2020-12-20