如何使用Numpy简化井字棋（Tic-Tac-Toe）胜负判断代码？

原代码通过硬编码索引逐一检查行、列、对角线，重复逻辑多，扩展或修改时容易出错。利用Numpy的数组操作结合3阶幻方，能大幅简化判断逻辑，提升代码可维护性。

核心思路 #

3阶幻方的每行、每列、两条对角线元素之和均为15。给井字棋每个位置对应幻方中的数值，当某玩家（X/O）占据的三个位置对应的幻方数值之和等于15时，说明该玩家连成一线，获得胜利。

实现步骤 #

1. 将一维棋盘数组转为3×3的Numpy二维数组，简化后续操作。
2. 定义3阶幻方矩阵，与棋盘位置一一对应。
3. 为X和O分别生成布尔掩码，筛选出各自占据的位置。
4. 计算每个玩家占据位置对应的幻方数值之和，判断是否等于15，同时通过步数判断（至少3步才可能赢）减少无效计算。
5. 直接返回获胜玩家，无赢家则返回None，避免使用全局变量。

简化后的代码 #

import numpy as np

def check_winner(board):
    # 把一维棋盘转为3x3的Numpy数组
    board_np = np.array(board).reshape(3, 3)
    # 3阶幻方矩阵，对应棋盘每个位置的数值
    magic_square = np.array([[8, 1, 6],
                             [3, 5, 2],
                             [4, 9, 7]])
    
    # 检查X是否获胜
    x_positions = magic_square[board_np == "X"]
    if len(x_positions) >= 3 and np.sum(x_positions) == 15:
        return "X"
    
    # 检查O是否获胜
    o_positions = magic_square[board_np == "O"]
    if len(o_positions) >= 3 and np.sum(o_positions) == 15:
        return "O"
    
    # 无获胜者
    return None

代码说明 #

• 幻方映射：所有获胜组合对应的幻方数值之和固定为15，无需逐一枚举行、列、对角线的索引。
• Numpy数组操作：利用布尔索引快速筛选玩家位置，代码更简洁，避免重复判断逻辑。
• 模块化设计：函数直接返回结果，没有依赖全局变量，更容易集成到其他逻辑中。

内容的提问来源于stack exchange，提问作者Xiaolin