写个贪吃蛇小游戏的代码

好的，以下是Python实现的简单贪吃蛇小游戏代码：

import pygame
import random

# 初始化pygame
pygame.init()

# 游戏窗口大小
WINDOW_WIDTH = 480
WINDOW_HEIGHT = 480

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

# 创建游戏窗口
window = pygame.display.set_mode((WINDOW_WIDTH, WINDOW_HEIGHT))
pygame.display.set_caption("Snake Game")

# 定义蛇的初始位置和大小
SNAKE_SIZE = 10
snake_x = WINDOW_WIDTH // 2 - SNAKE_SIZE
snake_y = WINDOW_HEIGHT // 2 - SNAKE_SIZE
snake_w = SNAKE_SIZE
snake_h = SNAKE_SIZE

# 定义蛇的移动速度
snake_speed_x = SNAKE_SIZE
snake_speed_y = 0

# 定义食物的初始位置和大小
FOOD_SIZE = 10
food_x = random.randint(0, WINDOW_WIDTH - FOOD_SIZE)
food_y = random.randint(0, WINDOW_HEIGHT - FOOD_SIZE)
food_w = FOOD_SIZE
food_h = FOOD_SIZE

# 定义游戏分数
score = 0

# 游戏主循环
while True:
    # 处理游戏事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            exit()

    # 获取键盘输入
    keys = pygame.key.get_pressed()

    # 根据键盘输入改变蛇的方向
    if keys[pygame.K_LEFT]:
        snake_speed_x = -SNAKE_SIZE
        snake_speed_y = 0
    elif keys[pygame.K_RIGHT]:
        snake_speed_x = SNAKE_SIZE
        snake_speed_y = 0
    elif keys[pygame.K_UP]:
        snake_speed_x = 0
        snake_speed_y = -SNAKE_SIZE
    elif keys[pygame.K_DOWN]:
        snake_speed_x = 0
        snake_speed_y = SNAKE_SIZE

    # 移动蛇的位置
    snake_x += snake_speed_x
    snake_y += snake_speed_y

    # 判断蛇是否吃到食物
    if snake_x < food_x + food_w and snake_x + snake_w > food_x and snake_y < food_y + food_h and snake_y + snake_h > food_y:
        # 食物被吃掉，重新生成食物
        food_x = random.randint(0, WINDOW_WIDTH - FOOD_SIZE)
        food_y = random.randint(0, WINDOW_HEIGHT - FOOD_SIZE)
        # 增加分数
        score += 10

    # 绘制游戏窗口
    window.fill(BLACK)
    pygame.draw.rect(window, GREEN, [food_x, food_y, food_w, food_h])
    pygame.draw.rect(window, BLUE, [snake_x, snake_y, snake_w, snake_h])
    pygame.display.update()

    # 判断蛇是否碰到边界
    if snake_x < 0 or snake_x > WINDOW_WIDTH - SNAKE_SIZE or snake_y < 0 or snake_y > WINDOW_HEIGHT - SNAKE_SIZE:
        # 游戏结束
        print("Game Over! Your score is", score)
        pygame.quit()
        exit()

    # 延时
    pygame.time.delay(100)

以上代码实现了简单贪吃蛇的基本功能，可以通过键盘控制蛇的方向，吃掉食物增加分数，碰到边界游戏结束。可以根据需求进行改进和完善。