Numpy中的if/else列表推导式的等价形式

8
有没有一种numpy的方法可以做到这一点?
n = [x-t if x > 0 else x for x in nps]

类似于这个。
n = np.array(a)
n[np.abs(n) < t] = 0

也许类似这样的东西吗?
n[n > 0] = n-t
1个回答

15

现在无法进行测试,但请尝试。

np.where(n > 0, n - t, n)

请参见文档

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接