Numpy相当于if / else列表理解

时间:2013-10-18 20:52:32

标签: python 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 个答案:

答案 0 :(得分:10)

现在无法测试,但请尝试

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

请参阅documentation