numpy数组值错误，在数组中查找特定元素

Question

我已经加载了一个数组形状的图像（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（）

有人知道怎么了吗？ （很抱歉，我也是初学者，也不是程序员。）