Pygame绘制rect跟随另一个rect基于他们的坐标(即蛇游戏)

import pygame
import sys

# 初始化Pygame
pygame.init()

# 设置窗口大小
screen_width, screen_height = 640, 480
screen = pygame.display.set_mode((screen_width, screen_height))
pygame.display.set_caption("Snake Game")

# 定义颜色
white = (255, 255, 255)
black = (0, 0, 0)

# 蛇头初始位置和大小
snake_head = pygame.Rect(screen_width // 2, screen_height // 2, 20, 20)
snake_body = []

# 移动速度
speed = [5, 0]

clock = pygame.time.Clock()

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

    # 获取键盘输入
    keys = pygame.key.get_pressed()
    if keys[pygame.K_LEFT]:
        speed = [-5, 0]
    elif keys[pygame.K_RIGHT]:
        speed = [5, 0]
    elif keys[pygame.K_UP]:
        speed = [0, -5]
    elif keys[pygame.K_DOWN]:
        speed = [0, 5]

    # 更新蛇头位置
    snake_head.move_ip(speed)

    # 防止蛇头移出屏幕
    if snake_head.left < 0:
        snake_head.left = screen_width
    elif snake_head.right > screen_width:
        snake_head.right = 0
    elif snake_head.top < 0:
        snake_head.top = screen_height
    elif snake_head.bottom > screen_height:
        snake_head.bottom = 0

    # 更新蛇身位置
    snake_body.insert(0, snake_head.copy())
    if len(snake_body) > 3:  # 控制蛇身长度
        snake_body.pop()

    # 绘制背景
    screen.fill(black)

    # 绘制蛇身
    for segment in snake_body:
        pygame.draw.rect(screen, white, segment)

    # 更新屏幕
    pygame.display.flip()

    # 控制帧率
    clock.tick(10)