linux 贪吃蛇终极脚本,用于深度学习的贪吃蛇游戏脚本

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