np.where

Creator

Created

2026 Jan 28 23:54

Editor

Edited

2026 Jan 28 23:59

Refs

1 argument → returns indices

3 arguments → selects values (ternary operator)


 np.where([True, False, True], [1,2,3], [4,5,6])  # [1, 5, 3]
  np.where(np.array([1,2,3]) > 1)  # (array([1, 2]),)


np.where(arr > 5, arr, 0)


 np.where(arr > 5)

////////