基本上，我有一个图像，我需要找到红色像素，并输出一个红色像素的图像，但在黑色背景上是白色的。我必须使用maplotlib，所以我首先使用plt.imread读取我的图像，并将其转换为一个numpy数组。主要问题是我不知道如何访问数组的第三维，RGB值本身所在的位置，并将它们赋给一个变量。
我的形象是这样的：image
我的输出应该是：output image
到目前为止，我已经编写了这个代码，但是我不知道我做错了什么。上限和下限阈值应该检测像素，所以如果“red”高于上限阈值，红色像素将被检测到。
我的嵌套for循环应该获取rgb_image的宽度和长度，并从中找到像素，但我很确定我没有正确使用它们。
任何帮助将不胜感激！

import numpy as np
import matplotlib.pyplot as plt

def find_red_pixels(map_filename,upper_threshold=100,lower_threshold=50):
    rgb_image = plt.imread(map_filename)

    for x in range(0,rgb_image[1]):
        for y in range(0,rgb_image[0]):
            rgb_image = rgb_image * 255
            red = rgb_image[...,0]
            green = rgb_image[...,1] = 0
            blue = rgb_image[...,2] = 0
            if (red > upper_threshold).any() and (green < lower_threshold).any() and (blue < lower_threshold).any():
                plt.imshow(red)
                plt.show()
                         
print(find_red_pixels('./data/map.png'))