日拱一卒，玩转伯克利Python入门项目 - 手把手带你写一个简单游戏AI

闲谈

2023-09-01 14:37:46

大家好，我是Coder梁，今天给大家分享的是伯克利的CS61A课程的第一个project——创建一个贪吃蛇游戏AI。虽然说是project，但其实代码量不大，难度也不高。只要你掌握了Python基础知识和人工智能算法，就可以轻松完成。

项目介绍

贪吃蛇是一个经典的游戏，相信大家都玩过。在这个游戏中，玩家控制一条蛇，通过吃掉食物来使它变长。当蛇头撞到墙或自身时，游戏结束。

我们的目标是创建一个AI，让它能够自动控制蛇的移动，并尽可能多地吃掉食物。

实现步骤

1. 导入必要的库

import pygame
import sys
import random

2. 初始化Pygame

pygame.init()

3. 设置屏幕大小

screen_width = 600
screen_height = 600
screen = pygame.display.set_mode((screen_width, screen_height))

4. 创建贪吃蛇

snake = [(200, 200), (200, 210), (200, 220)]

5. 创建食物

food = (300, 300)

6. 游戏循环

while True:
    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP:
                snake_direction = "up"
            elif event.key == pygame.K_DOWN:
                snake_direction = "down"
            elif event.key == pygame.K_LEFT:
                snake_direction = "left"
            elif event.key == pygame.K_RIGHT:
                snake_direction = "right"

    # 移动贪吃蛇
    snake_head = snake[0]
    if snake_direction == "up":
        snake_head = (snake_head[0], snake_head[1] - 10)
    elif snake_direction == "down":
        snake_head = (snake_head[0], snake_head[1] + 10)
    elif snake_direction == "left":
        snake_head = (snake_head[0] - 10, snake_head[1])
    elif snake_direction == "right":
        snake_head = (snake_head[0] + 10, snake_head[1])

    # 将新蛇头添加到贪吃蛇中
    snake.insert(0, snake_head)

    # 如果蛇头撞到墙或自身，游戏结束
    if snake_head[0] < 0 or snake_head[0] >= screen_width or snake_head[1] < 0 or snake_head[1] >= screen_height or snake_head in snake[1:]:
        pygame.quit()
        sys.exit()

    # 如果蛇头吃到食物，食物的位置随机改变，并增加蛇的长度
    if snake_head == food:
        food = (random.randint(0, screen_width - 10), random.randint(0, screen_height - 10))
        snake.append((snake_head[0], snake_head[1]))

    # 清除屏幕
    screen.fill((0, 0, 0))

    # 绘制贪吃蛇
    for snake_part in snake:
        pygame.draw.rect(screen, (255, 255, 255), (snake_part[0], snake_part[1], 10, 10))

    # 绘制食物
    pygame.draw.rect(screen, (255, 0, 0), (food[0], food[1], 10, 10))

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

    # 设置帧速率
    pygame.time.Clock().tick(10)