Python：列表列表__getitem__

Question

我已經為連接4游戲創建了一個Board類，它使用了一個列表列表。 我想讓對象支持索引板的行和列：

class Test:
    def __init__(self, cols, rows):
        self.cols = cols
        self.rows = rows
        self.data = [[' '] * cols for row in range(rows)]

    def __iter__(self):
        yield from self.data

    def __getitem__(self, row):
        pass
        # what should this be?

我知道我需要定義__getitem__但是這個方法應該允許什么語法如b[1][1] ？ 我已經看了一些過去的問題，但我似乎無法找到我正在尋找的答案。

編輯 ：也許以下是正確的做事方式？

def __getitem__(self, row):
    return self.data[row]

Answer 1

你需要理解的基本事情是，對於像b[1][2]這樣的語法，有兩個 __getitem__調用。 首先，調用b.__getitem__(1) ，並返回一個對象。 然后，調用returned_object.__getitem__(2) 。 因此，無論什么類別b都不能完全控制整個過程。

在上面的示例中，如果您將其定義為

def __getitem__(self, row):
    return self.data[row]

然后b[1][2]將首先調用b.__getitem__(1) ，這將返回self.data[1] ，這將是一個列表。 該列表也支持__getitem__ ，因此將調用下一個list.__getitem__(2) 。

如果你想控制這兩個 ，那么你需要定義另一個表示列的類，父類需要返回其中一個對象而不是原始列表，然后那個可以為自定義功能實現自己的__getitem__ 。

其他選擇是支持b[1,2]這樣合法的語法 - 它會將(1,2)作為元組調用b.__getitem__((1,2)) （你可以把任何東西放在那里），但是它不那么自然。

Answer 2

您可以返回子列表所代表的row

def __getitem__(self, row):
    return self.data[row]

例如

>>> t = Test(2,5)
>>> t.data
[[' ', ' '],
 [' ', ' '],
 [' ', ' '],
 [' ', ' '],
 [' ', ' ']]

排好隊

>>> t[1]
[' ', ' ']

得到一個元素

>>> t[1][0]
' '

請注意，上面示例中的第二個操作[0]不需要您定義它，因為它只是從list調用__getitem__ ，因為它就是這樣。

Answer 3

只需返回行：

def __getitem__(self, row):
    return self.data[row]