写一个贪吃蛇的游戏

以下是一个简单的贪吃蛇游戏的Python代码：

import pygame
import random

# 初始化游戏
pygame.init()

# 设置游戏窗口的大小
window_width = 800
window_height = 600
window = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption("贪吃蛇游戏")

# 定义颜色
black = pygame.Color(0, 0, 0)
white = pygame.Color(255, 255, 255)
red = pygame.Color(255, 0, 0)
green = pygame.Color(0, 255, 0)
blue = pygame.Color(0, 0, 255)

# 定义蛇的初始位置和移动的速度
snake_position = [100, 50]
snake_body = [[100, 50], [90, 50], [80, 50]]
snake_speed = pygame.time.Clock()

# 定义食物的初始位置
food_position = [random.randrange(1, (window_width//10)) * 10, random.randrange(1, (window_height//10)) * 10]
food_spawn = True

# 定义初始方向和改变的方向
direction = 'RIGHT'
change_to = direction

# 定义游戏结束的标志
game_over = False

# 定义蛇的移动函数
def change_direction():
    global direction, change_to
    if change_to == 'RIGHT' and not direction == 'LEFT':
        direction = 'RIGHT'
    if change_to == 'LEFT' and not direction == 'RIGHT':
        direction = 'LEFT'
    if change_to == 'UP' and not direction == 'DOWN':
        direction = 'UP'
    if change_to == 'DOWN' and not direction == 'UP':
        direction = 'DOWN'

# 定义游戏结束的函数
def game_over_screen():
    font = pygame.font.SysFont(None, 72)
    text = font.render("游戏结束", True, red)
    window.blit(text, (window_width/2 - text.get_width()/2, window_height/2 - text.get_height()/2))
    pygame.display.flip()
    pygame.time.wait(2000)
    pygame.quit()

# 游戏主循环
while not game_over:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            game_over = True
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_RIGHT:
                change_to = 'RIGHT'
            if event.key == pygame.K_LEFT:
                change_to = 'LEFT'
            if event.key == pygame.K_UP:
                change_to = 'UP'
            if event.key == pygame.K_DOWN:
                change_to = 'DOWN'

    # 检查蛇是否撞到墙壁
    if snake_position[0] >= window_width or snake_position[0] < 0 or snake_position[1] >= window_height or snake_position[1] < 0:
        game_over = True
        game_over_screen()

    # 检查蛇是否撞到自己
    for block in snake_body[1:]:
        if snake_position[0] == block[0] and snake_position[1] == block[1]:
            game_over = True
            game_over_screen()

    # 检查蛇是否吃到食物
    if snake_position[0] == food_position[0] and snake_position[1] == food_position[1]:
        food_spawn = False
    else:
        snake_body.pop()

    if not food_spawn:
        food_position = [random.randrange(1, (window_width//10)) * 10, random.randrange(1, (window_height//10)) * 10]
        food_spawn = True

    # 更新蛇的位置
    if direction == 'RIGHT':
        snake_position[0] += 10
    if direction == 'LEFT':
        snake_position[0] -= 10
    if direction == 'UP':
        snake_position[1] -= 10
    if direction == 'DOWN':
        snake_position[1] += 10

    # 更新蛇的身体
    snake_body.insert(0, list(snake_position))

    # 渲染游戏窗口
    window.fill(black)
    for block in snake_body:
        pygame.draw.rect(window, green, pygame.Rect(block[0], block[1], 10, 10))
    pygame.draw.rect(window, white, pygame.Rect(food_position[0], food_position[1], 10, 10))

    # 更新游戏窗口
    pygame.display.update()

    # 控制游戏速度
    snake_speed.tick(15)

# 退出游戏
pygame.quit()

在运行此代码之前，需要确保已经安装了pygame库。可以使用以下命令安装：

pip install pygame

此代码实现了一个简单的贪吃蛇游戏，在窗口中绘制了蛇和食物，并实现了蛇的移动、食物的生成、游戏结束等功能。玩家可以通过键盘上的箭头键来控制蛇的移动方向。游戏结束后，会显示一个"游戏结束"的提示，并在2秒后退出游戏