如何合并Board类的mark_block与mark_cell函数？能否不用if语句实现？

带条件判断的合并方式 #

最简单的整合方式是将两个方法合并为一个mark方法，通过可选参数区分操作目标：

import numpy as np

class Board:
    def __init__(self):
        self.cells = np.zeros((3, 3, 3, 3))
        self.block = np.zeros((3, 3))

    def mark(self, main_row, main_col, player, row=None, col=None):
        if row is not None and col is not None:
            self.cells[main_row][main_col][row][col] = player
        else:
            self.block[main_row][main_col] = player

调用逻辑和原方法完全一致：

• 标记区块：board.mark(main_row, main_col, player)
• 标记单元格：board.mark(main_row, main_col, player, row, col)

无if语句的实现方式 #

如果要完全避免if判断，可以利用可变参数和数组索引特性实现：

import numpy as np

class Board:
    def __init__(self):
        self.cells = np.zeros((3, 3, 3, 3))
        self.block = np.zeros((3, 3))
        # 按索引对应操作目标：0对应block，1对应cells
        self._targets = (self.block, self.cells)

    def mark(self, main_row, main_col, player, *args):
        # 根据可变参数长度自动选择目标数组
        target = self._targets[len(args) // 2]
        # 拼接索引并完成赋值
        target[(main_row, main_col) + args] = player

核心逻辑说明：

• 用*args接收可选的row和col参数
• 未传入row/col时，args为空元组，len(args)//2=0，自动选择self.block，索引为(main_row, main_col)
• 传入row/col时，args长度为2，len(args)//2=1，自动选择self.cells，索引为(main_row, main_col, row, col)

调用方式：

• 标记区块：board.mark(0, 1, 1)
• 标记单元格：board.mark(0, 1, 1, 2, 0)

内容的提问来源于stack exchange，提问作者Edward Henry Brenner