import numpy as np
x = np.array([[1, 1], [2, 1], [2, 2], [1, 2], [0, 1]])
np.all(x == [0, 1], axis=1)
np.where(np.all(x == [0, 1], axis=1))
np.where(np.any(x != [0, 1], axis=1))
Python/Numpy индекс массива в массиве