[Numpy-discussion] Getting indices from numpy array with condition