用于深度學(xué)習(xí)的貪吃蛇游戲腳本

功能:

  1. 上下左右鍵控制貪吃蛇朝各個(gè)方向移動(dòng),每吃一個(gè)食物就長(zhǎng)大一截;
  2. 黑底白框易于機(jī)器識(shí)別;
  3. 機(jī)器實(shí)現(xiàn)自動(dòng)化時(shí),可以深度游戲底層,自創(chuàng)user event事件,用pygame.event.post提交鍵盤事件。
  4. 空格鍵暫停,esc鍵退出。

代碼:

import pygame
import sys,os,random

pygame.init()
screen = pygame.display.set_mode((600,600))
pygame.display.set_caption('巴拉巴拉小魔仙')
fclock = pygame.time.Clock()
snake = [[300,300],[290,300],[280,300]]
directions = {'up':(0,-10),'down':(0,10),'left':(-10,0),'right':(10,0)}
direct = 'right'
pause = True
fps = 10

def manhattan_distance(p1,p2):
    return abs(p1[0]-p2[0]) + abs(p1[1]-p2[1])

def gen_food_pos():
    x,y = random.randint(0,590),random.randint(0,590)
    if (x,y) in snake:
        gen_food_pos()
    return x,y

def gen_food(food_pos):
    return pygame.draw.rect(screen, (255,255,255), [food_pos[0],food_pos[1], 10, 10])

def gen_snake(snake):
    snake_body = []
    for x,y in snake:
        snake_body.append(pygame.draw.rect(screen, (255,255,255), [x, y, 10, 10]))
    return snake_body

def move(dircetion):
    step = directions.get(dircetion)
    snake[1:] = snake[:-1]
    snake[0] = snake[0][0] + step[0] ,snake[0][1] + step[1] 
    return snake

def inhance_snake(pos,snake):
    head = snake_body[0]
    if pos[0] < head.x :
        if abs(pos[0] - head.x) > abs(pos[1] - head.y) :
            newhead = [[head.x-10,head.y]]
            snake = newhead + snake
        else:
            newhead = [[head.x+10,head.y]]
            snake = newhead + snake
    else :
        if abs(pos[1] - head.y) > abs(pos[0] - head.x) :
            newhead = [[head.x-10,head.y-10]]
            snake = newhead + snake
        else:
            newhead = [[head.x-10,head.y+10]]
            snake = newhead + snake
    return snake


while True:
    food_pos = gen_food_pos()
    while True:
        if food_pos != snake [0]:
            for event in pygame.event.get():
                if   event.type == pygame.QUIT:sys.exit()
                elif event.type == pygame.KEYDOWN:
                    if   event.key == pygame.K_UP   :direct = 'up'
                    elif event.key == pygame.K_DOWN :direct = 'down'
                    elif event.key == pygame.K_LEFT :direct = 'left'
                    elif event.key == pygame.K_RIGHT:direct = 'right'
                    elif event.key == pygame.K_ESCAPE:sys.exit()
                    elif event.key == pygame.K_SPACE and pause is True:
                        directions = {'up':(0,0),'down':(0,0),'left':(0,0),'right':(0,0)}
                        pause = False
                    elif event.key == pygame.K_SPACE and pause is False:
                        directions = {'up':(0,-10),'down':(0,10),'left':(-10,0),'right':(10,0)}
                        pause = True
                    else :pass
            
            fclock.tick(fps)
            screen.fill((0,0,0))
            food = gen_food(food_pos)
            snake = move(direct)
            snake_body = gen_snake(snake)
            pygame.display.update()

            head = snake_body[0]
            if manhattan_distance(food_pos,(head.x,head.y)) <= 15:
                snake = inhance_snake(food_pos,snake)
                break

可以改善的地方:

  1. 增加積分制,作為深度學(xué)習(xí)的激勵(lì)函數(shù);
  2. 沒(méi)有實(shí)現(xiàn)小蛇的懲罰項(xiàng),目前無(wú)論是碰到四壁還是自身都不會(huì)死的。

樣本:

巴拉巴拉小魔仙貪吃蛇
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容