如何在pygame中删除图片中特定颜色并将图片中的颜色设置为透明？

import pygame
import sys

pygame.init()

screen = pygame.display.set_mode((800, 600))
image = pygame.image.load("image.png")

image_data = pygame.PixelArray(image)

color_to_remove = pygame.Color(255, 0, 0)  # 要删除的颜色，这里以红色为例
transparent_color = pygame.Color(0, 0, 0, 0)  # 设置为透明的颜色

for x in range(image.get_width()):
    for y in range(image.get_height()):
        if image_data[x, y] == color_to_remove:
            image_data[x, y] = transparent_color

new_image = pygame.Surface((image.get_width(), image.get_height()), pygame.SRCALPHA)
new_image.blit(image, (0, 0))

del image
del image_data

screen.blit(new_image, (0, 0))
pygame.display.flip()

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()