用python代码做一个贪吃蛇的小游戏

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

import pygame
import random

# 定义颜色
WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
GREEN = (0, 255, 0)
RED = (255, 0, 0)

# 定义贪吃蛇类
class Snake:
    def __init__(self):
        self.size = 10
        self.x = 0
        self.y = 0
        self.dx = self.size
        self.dy = 0
        self.body = []
        self.length = 1

    def update(self):
        # 移动蛇头
        self.x += self.dx
        self.y += self.dy

        # 检查是否吃到食物
        if self.x == food.x and self.y == food.y:
            self.length += 1
            food.spawn()
        else:
            # 移动蛇身
            if len(self.body) >= self.length:
                self.body.pop(0)
            self.body.append((self.x, self.y))

    def draw(self, surface):
        # 绘制蛇头
        pygame.draw.rect(surface, GREEN, (self.x, self.y, self.size, self.size))

        # 绘制蛇身
        for x, y in self.body:
            pygame.draw.rect(surface, GREEN, (x, y, self.size, self.size))

    def move_left(self):
        self.dx = -self.size
        self.dy = 0

    def move_right(self):
        self.dx = self.size
        self.dy = 0

    def move_up(self):
        self.dx = 0
        self.dy = -self.size

    def move_down(self):
        self.dx = 0
        self.dy = self.size

# 定义食物类
class Food:
    def __init__(self):
        self.size = 10
        self.x = 0
        self.y = 0

    def spawn(self):
        # 随机生成食物位置
        self.x = random.randint(0, (WIDTH - self.size) // self.size) * self.size
        self.y = random.randint(0, (HEIGHT - self.size) // self.size) * self.size

    def draw(self, surface):
        pygame.draw.rect(surface, RED, (self.x, self.y, self.size, self.size))

# 初始化 pygame
pygame.init()

# 定义窗口大小
WIDTH = 500
HEIGHT = 500

# 创建窗口
display = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption('贪吃蛇')

# 创建贪吃蛇和食物对象
snake = Snake()
food = Food()

# 生成初始食物位置
food.spawn()

# 游戏循环
done = False
clock = pygame.time.Clock()
while not done:
    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            done = True
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                snake.move_left()
            elif event.key == pygame.K_RIGHT:
                snake.move_right()
            elif event.key == pygame.K_UP:
                snake.move_up()
            elif event.key == pygame.K_DOWN:
                snake.move_down()

    # 更新贪吃蛇和食物
    snake.update()

    # 绘制背景
    display.fill(BLACK)

    # 绘制贪吃蛇和食物
    snake.draw(display)
    food.draw(display)

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

    # 控制帧率
    clock.tick(10)

# 退出 pygame
pygame.quit()

这个代码示例使用了 pygame 库来创建游戏窗口和处理事件、绘制图形等。运行代码可以看到一个简单的贪吃蛇游戏，通过上下左右键控制蛇的移动并吃掉食物，当蛇的长度达到一定程度时游戏结束。