numpy_array = numpy.arange(9).reshape((3, 3))
confused_array = numpy.arange(9).reshape((3, 3)) % 2
mask = numpy.in1d(numpy_array, repeat_set).reshape(numpy_array.shape)
mask
~mask
numpy_array == 0
numpy_array != 0
confused_array[mask] = 1
confused_array[~mask & (numpy_array == 0)] = 0
confused_array[~mask & (numpy_array != 0)] = 2
confused_array
Проверьте, находятся ли значения в наборе в массиве numpy в python