python贪吃蛇核心功能实现下

#创建一条新的蛇
def create_new_snake():
    snake_body = []
    snake_x = random.randint(5, rows - 5)
    snake_y = random.randint(5, columns - 5)
    snake_body.append([snake_x, snake_y])
    snake_body.append([snake_x, snake_y + 1])
    snake_body.append([snake_x, snake_y + 2])
    return snake_body

snake = create_new_snake()

def create_new_apple(snake_body):
    apple_in_game = False
    while not apple_in_game:
        apple_x = random.randint(0, rows - 1)
        apple_y = random.randint(0, columns - 1)
        # print(apple_x, apple_y)
        if (apple_x, apple_y) not in snake_body:
            new_apple = pygame.image.load('apple.png')
            new_apple = pygame.transfORM.scale(new_apple, (tile_size - 2, tile_size - 2))
            apple_x_location = apple_y * tile_size + x_margin/2
            apple_y_location = apple_x * tile_size + y_margin/2

            display.blit(new_apple, (apple_x_location, apple_y_location))
            apple_in_game = True
    return new_apple, (apple_x, apple_y), apple_x_location, apple_y_location

if __name__ == '__main__':
    pygame.init()
    snake = create_new_snake()
    # 未按下按键时，默认的初识移动方向
    direction = directions[random.randint(0, 3)]
    # 创建一个小苹果
    apple, apple_location, apple_left, apple_top = create_new_apple(snake)
    # 吃到苹果标志
    apple_was_eaten = False

        if apple_location in snake:
            apple, apple_location, apple_left, apple_top = create_new_apple(snake)
            apple_was_eaten = True
            my_score += 1