檢查tic-tac-toe的優勝者 - NumPy/Python

Question

我想檢查井字遊戲中所有可能的贏家條件，我如何以功能性方式重寫？

board = numpy.array([[0, 0, 0], 
        [0, 0, 0], 
        [0, 0, 0]]) 
player = 1 

if any([(board[0, :] == player).all(), 
     (board[:, 0] == player).all(), 
     (board[1, :] == player).all(), 
     (board[:, 1] == player).all(), 
     (board[2, :] == player).all(), 
     (board[:, 2] == player).all()]): 
    print('Win') 

Answer 1

使用的any和all組合 -

mask = board==player 
out = mask.all(0).any() | mask.all(1).any() 

爲了說明diagonal的，多做一些工作 -

out |= np.diag(mask).all() | np.diag(mask[:,::-1]).all()