numpy get index of condition code example

Example: numpy argwhere

test = np.array([-2,3,4,-8,1])
np.argwhere(test < 0)

>>>
array([[0],
       [3]], dtype=int64)