Ошибка значения Numpy Array, при поиске определенного элемента в массиве - PullRequest
Новая
       12

Ошибка значения Numpy Array, при поиске определенного элемента в массиве

0 голосов
/ 14 ноября 2018

Я загрузил изображение в виде массива (1024, 1152, 3).Затем преобразуем его в 2D: 

x, y, z = img.shape
img_2d = img.reshape(x*y,z)

Во второй строке содержатся значения системы RGB (например, print(img_2d[70000,:]) дает вывод [171 135 115]. Теперь я хочу назначить каждый код цвета RGB одному целому числу (например,[255 0 0] = 1 для красного) и, следовательно, итерация массива с помощью цикла: 

for i in img_2d:
     if img_2d[i,0] == 255:
            print(img_2d[i,1])

Однако я получаю ту загадочную ошибку, которую не понимаю:

ValueError: Значение истинности массива с более чем одним элементом неоднозначно. Используйте a.any () или a.all ()

Любой, кто знает, что случилось? (Извините, но я новичок, а не программист.)

...