基于python做一个2048小游戏

def mergeLst(lst):
    lst = [x for x in lst if x] # 删除lst中的0
    newLst = []
    N, i = len(lst), 0
    while i < N:
        if i<N-1 and lst[i] == lst[i+1]:
            newLst.append(lst[i]*2)
            i += 2
        else:
            newLst.append(lst[i])
            i += 1
    return newLst + (4-len(newLst))*[0]

mergeLst([0,2,2,0]) # [4, 0, 0, 0]
mergeLst([2,2,2,2]) # [4, 4, 0, 0]
mergeLst([2,2,8,4]) # [4, 8, 4, 0]

[mergeLst(x) for x in mat]

def rotate(mat):
    newMat = [[[] for _ in mat] for b in mat[0]]
    for i,row in enumerate(mat, 1):
        for j,r in enumerate(row, 0):
            newMat[j][-i] = r
    return newMat

KEY_DCT = {
    pygame.K_LEFT:0,   pygame.K_DOWN:1,
    pygame.K_RIGHT:2,  pygame.K_UP:3
}

def updateMat(mat, key):
    N = KEY_DCT[key]
    for _ in range(N):
        mat = rotate(mat)
    mat = [mergeLst(x) for x in mat]
    for _ in range(4-N):
        mat = rotate(mat)   # 旋转回来
    return mat

from itertools import product
from random import sample, randint
def addNew(mat):
    ijs = []
    for i,j in product(range(4), range(4)):
        if mat[i][j] == 0:
            ijs.appen((i,j))
    if len(ijs) == 0:
        return False
    i, j = sample(ijs, 1)[0]   # 挑选一个不为0的点
    x = randint(1,100)
    x = 7 - np.floor(np.log2(x))
    mat[i,j] = int(2**x)
    return True

GRAY = (205, 205, 205)
WHITE = (255, 255, 255)
GREEN = (0, 255, 0)
SIZE = 80

def setColor(N):
    if N == 0:
        return (233, 233, 233)
    n = int(math.log2(N))
    delta = int(255/15)
    return (255-n*delta, n*delta, (128+n*delta)%255)


def draw(win, font, mat):
    for i,j in product(range(4), range(4)):
        ch = str(mat[i][j])
        rect = (j*SIZE, i*SIZE, SIZE, SIZE)
        c = setColor(mat[i][j])
        pygame.draw.rect(win, c, rect)
        pygame.draw.rect(win, GRAY, rect, 5)

        txt = font.render(ch, True, GREEN)
        offset =  (0.5 - len(ch)/10)*SIZE
        xy = (j*SIZE+offset, (i+0.3)*SIZE)
        win.blit(txt, xy)

def gui():
    pygame.init()
    win = pygame.display.set_mode(
        (4 * SIZE, 4 * SIZE))
    pygame.display.set_caption("迷宫游戏")
    font.init()
    f = font.Font('C:\\WINDOWS\\Fonts\\ARLRDBD.TTF', 32)
    running = True

    mat = [[0]*4 for _ in range(4)]
    addNew(mat)
    while running:
        win.fill(WHITE)
        for evt in pygame.event.get():
            if evt.type == pygame.QUIT:
                return
            if evt.type == pygame.KEYDOWN and evt.key in KEY_DCT:
                newMat = updateMat(mat, evt.key)
                if newMat!=mat:
                    mat = newMat
                    running = addNew(mat)
        draw(win, f, mat)
        # 判断玩家是否到达出口
        if max([max(x) for x in mat])==2048:
            print("恭喜通关")
            running = False
        if min([min(x) for x in mat])>0:
            print("游戏失败")
            running = False

        pygame.display.update()