Как сравнить, если многомерные элементы массива присутствуют в другом массиве с Numpy? - PullRequest
0 голосов
/ 04 ноября 2019

В основном я ищу решение, похожее на numpy, но любое другое эффективное решение приветствуется.

Я хочу определить, присутствуют ли цвета пикселей изображения или нет в другом массиве, в котором хранится набор цветов RGB. У меня есть цвета пикселей, уложенные в массив rgb_img (1, n, 3), и у меня есть матрица set_color (1, n, 3).

color_set = np.array([[[ 75,  99, 147],
                       [ 75,  98, 148],
                       [ 75,  97, 154],
                       [ 75,  97, 156],
                       [ 75,  97, 157]]], dtype=np.uint8)

rgb_img = np.array([[[ 2,  3,  4],
                     [ 9, 11, 15],
                     [11, 14, 19],
                     [75,  97, 155],
                     [75,  97, 156],
                     [ 5,  6,  9]]], dtype=np.uint8)

Желаемый результат:

[False, False, False, False, True, False]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...