python - python 上的 tic tac toe

Question

我正在尝试找出一个 for 循环和 if 语句来检查 tic tac toe 游戏中的获胜者。我需要这个，因为我有一个 connect 4 游戏，并且我计划使用相同的伪代码。

这是我到目前为止所拥有的:

col_size = 3
        row_size = 3
        ttt = list()
        for n in range(col_size):
            rows = input(msg[n]+ ":")
            ttt.append(rows)
        print(ttt)
        row_ttt= list()
        for i in range(row_size):
            one_row = list()
            for j in range(col_size):
                one_row+= ttt[i][j]
            row_ttt.append(one_row)
        diagonal_ttt= list()
        for j in range(col_size):
            one_diagonal= ""
            for i in range(row_size):
                one_diagonal+= ttt[i][j+1:2]
            diagonal_ttt.append(one_diagonal)
        print(row_ttt)
        valid_symbols = ['x','X','o','O','.']

            for j in range(col_size):
                for i in range(row_size):
                    if row_ttt[i][j]== row_ttt[i:i+1][j] == row_ttt[i:i+2][j]:
                        #print(row_ttt[i][j]) to check the element
                        #print(row_ttt[i:i+1][j]) to check element
                        #print(row_ttt[i:i+2][j]) to check element
                        valid = True
                print("valid board - " + ttt[i][j] + " is the winner")
                break

我在这里手动写了它，但我想要更短的代码:

            if ttt[0][0] == ttt[1][1] == ttt[2][2]:
                print("valid board " + ttt[0][0] + " is the winner")
                break
            if ttt[0][2] == ttt[1][1] == ttt[2][0]:
                print("valid board " + ttt[2][0] + " is the winner")
                break
            if ttt[0][0] == ttt[0][1] == ttt[0][2]:
                print("valid board " + ttt[0][0] + " is the winner")
                break
            if ttt[1][0] == ttt[1][1] == ttt[1][2]:
                print("valid board " + ttt[0][0] + " is the winner")
                break
            if ttt[2][0] == ttt[2][1] == ttt[2][2]:
                print("valid board " + ttt[0][0] + " is the winner")
                break
            if ttt[0][0] == ttt[1][1] == ttt[2][2]:
                print("valid board " + ttt[0][0] + " is the winner")
                break
            if ttt[0][0] == ttt[1][0] == ttt[2][0]:
                print("valid board " + ttt[0][0] + " is the winner")
                break
            if ttt[0][1] == ttt[1][1] == ttt[2][1]:
                print("valid board " + ttt[0][1] + " is the winner")
                break
            if ttt[0][2] == ttt[1][2] == ttt[2][2]:
                print("valid board " + ttt[0][2] + " is the winner")
                break

Answer 1

>>> import numpy
>>> game_board = [[0,0,0],[0,1,2],[1,2,1]]
>>> #let 1 be X's and 2 be O's .... 0 is empty
>>> def check_win(board,player):
...     b= numpy.array(board)
...     x= b == player
...     return numpy.hstack([x.all(0),x.all(1),numpy.diag(x).all(),numpy.diag(x[:,::-1]).all()]).any()
...
>>> print check_win(game_board,1) #check for X win
>>> print check_win(game_board,2) # check for O win