Numpy.where использует

Question

Используйте numpy.where, чтобы получить все (R, G, B) в массиве numpy.ar с определенными значениями R, G и B

Проблема в том, что я не уверен, что могу использовать numpy.где получить то, что я хочу:

я попробовал следующий код:

L = numpy.array([[1,2,3],[1,1,1],[1,1,1]])
print(numpy.where(L==(1,1,1)))

(массив ([0, 1, 1, 1, 2, 2, 2], dtype = int64), массив ([0, 0, 1, 2, 0, 1, 2], dtype = int64))

и я понимаю, что он возвращает менякоординаты каждого элемента == 1, но я бы хотел, чтобы он возвращал индекс в L элемента, равный (1,1,1):

array ([1,2])

Answer 1

Вы ищете numpy.nonzero вместе с np.all (чтобы убедиться, что каждый из RGB соответствует):

>>> numpy.nonzero(numpy.all(L == (1, 1, 1), axis=1))[0]
array([1, 2])