如何在保留一种颜色的情况下将RGB图像转换为灰度图像？

import numpy as np
import cv2

def rgb_to_gray(image):
    # 获取图像的宽度和高度
    height, width, _ = image.shape
    
    # 创建一个与原图像大小相同的空灰度图像
    gray_image = np.zeros((height, width), dtype=np.uint8)
    
    # 将每个像素的RGB值转换为灰度值
    for y in range(height):
        for x in range(width):
            r, g, b = image[y, x]
            gray_value = int(0.2989 * r + 0.587 * g + 0.114 * b)
            gray_image[y, x] = gray_value
    
    return gray_image

# 读取RGB图像
image = cv2.imread("rgb_image.jpg")

# 将RGB图像转换为灰度图像
gray_image = rgb_to_gray(image)

# 显示灰度图像
cv2.imshow("Gray Image", gray_image)
cv2.waitKey(0)
cv2.destroyAllWindows()