Skip to content Skip to sidebar Skip to footer

Numpy Equivalent Of If/else List Comprehension

Is there a numpy way of doing n = [x-t if x > 0 else x for x in nps] similar to this n = np.array(a) n[np.abs(n) < t] = 0 something like this perhaps? n[n > 0] = n-t

Solution 1:

Can't test now, but try

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

See documentation


Post a Comment for "Numpy Equivalent Of If/else List Comprehension"