定义一个类中iter的多个方差

Question

我正在写一个连接4的游戏，有一个网格类，类似于游戏所使用的二维网格。我的网格类的底层数据结构是一个实例变量self.grid——一个二维列表。你知道吗

我重写了__iter__方法，如下所示：

def __iter__(self):
    for row in range(self.numRows):
        for col in range(self.numCols):
            yield self.grid[row][col]

我可以称之为：

for cell in grid:
    # do something with yielded cell

在某些情况下，我需要迭代指定的行或列。而不是写作

col = 0
for row in range(grid.numRows):
    # do something with grid.grid[row][col]

有没有什么通俗易懂的方法来定义第二个和第三个__iter__方法：

def __iter__(self, col)
    for row in range(self.numRows):
        yeild self.grid[row][col]

我们称之为：

for row in grid(col=0):
    # do something with the yielded cell

然后是相同的，但是有一个指定的row？你知道吗

Answer 1

你当然可以有很多这样的方法。您不能全部调用它们__iter__()，因为Python不支持方法重载。例如，您可以这样做：

def rows(self, col):
    for row in range(self.numRows):
        yield self.grid[row][col]

def cols(self, row):
    for col in range(self.numCols):
        yield self.grid[row][col]

def cells(self):
    for row in range(self.numRows):
        for col in range(self.numCols):
            yield self.grid[row][col]

然后你会写下

for row in grid.rows(col=0):
    # Do something