__getitem__在列表列表上有切片

Question

我正在创建一个代表列表列表的类。 __getitem__使我头疼。 一切都进行得很顺利，直到我引入slice作为参数为止。

示范代码

# Python version 2.7.5

class NestedLists:
   _Cells = [['.', '.', '.', '.', '.'],
             ['.', '.', 'N', '.', '.'],
             ['.', 'C', 'A', 'T', '.'],
             ['.', '.', 'P', '.', '.'],
             ['.', '.', '.', '.', '.']]

   def __getitem__(self, index):
      if isinstance(index, int):
         return self._Cells[index]
      elif isinstance(index, slice):
         return self._Cells[index]
      else:
         raise TypeError, "Invalid argument type"

nested = NestedLists()

print "Expecting A"
print nested[2][2]

print "Expecting CAT"
print nested[2][1:4]

print "Expecting ..N.."
print "          .CAT."
print "          ..P.."
print nested[1:4]

print "Expecting .N."
print "          CAT"
print "          .P."
print nested[1:4][1:4]

输出如下

Expecting A
A
Expecting CAT
['C', 'A', 'T']
Expecting ..N..
          .CAT.
          ..P..
[['.', '.', 'N', '.', '.'], ['.', 'C', 'A', 'T', '.'], ['.', '.', 'P', '.', '.']]
Expecting .N.
          CAT
          .P.
[['.', 'C', 'A', 'T', '.'], ['.', '.', 'P', '.', '.']]

显然，正在发生的事情是第二个[]运算符调用被应用到第一个[...]运算符的调用，但仍停留在最外层列表的上下文中。 解决方案使我难以理解。

Answer 1

您可能希望将用于多维访问的语法从obj[row][col]更改为使用单个元组索引： obj[row, col] 。 这是numpy的ndarray类型使用的格式，它非常有用，因为它使您可以立即查看索引的所有ndarray 。 您可以编写一个__getitem__ ，它可以在任何维度上切片：

def __getitem__(self, index):
   row, col = index
   if isinstance(row, int) and isinstance(col, (int, slice)):
      return self._Cells[row][col]
   elif isinstance(row, slice) and isinstance(col, (int, slice)):
      return [r[col] for r in self._Cells[row]]
   else:
      raise TypeError, "Invalid argument type"

Answer 2

当index是切片时，从__getitem__返回的是列表列表：

[['.', '.', 'N', '.', '.'], ['.', 'C', 'A', 'T', '.'], ['.', '.', 'P', '.', '.']]

第二个切片索引到此列表中，而不是将该切片应用于列表中的每个项目。

为了获得您要查找的行为，可以使NestedLists.__getitem__返回一个新类， NestedListsSlice index为切片时的NestedListsSlice 。 然后，可以定义自己的__getitem__来将第二个索引应用于：

class NestedLists:
    ...
    def __getitem__(self, index):
        ...
        elif isinstance(index, slice):
            return NestedListsSlice(self._Cells[index])
        ...

class NestedListsSlice():
    def __init__(self, _Cells):
        self._Cells = _Cells
    def __getitem__(self, index):
        if isinstance(index, int) or isinstance(index, slice):
            return list(x[index] for x in self._Cells)
        else:
            raise TypeError