Python中二维列表如何获取子区域元素的组成

2019-11-25 16:23:44

字体：大中小

来源：转载

供稿：网友

用过NumPY的应该都知道，在二维数组中可以方便地使用区域切片功能，如下图：

而这个功能在Python标准库的List中是不支持的，在List中只能以一维方式来进行切片操作：

但有时候我只想用一下这个功能，但又不想引入NumPY。其实这时候我也是可以在Python中实现的。这时候，只需在一个类中实现__getitem__特殊方法：

class Array: """实现__getitem__，支持序列获取元素、Slice等特性""" def __init__(self, lst):  self.__coll = lst def __repr__(self):  """显示列表"""  return '{!r}'.format(self.__coll) def __getitem__(self, key):  """获取元素"""  slice1, slice2 = key  row1 = slice1.start  row2 = slice1.stop  col1 = slice2.start  col2 = slice2.stop  return [self.__coll[r][col1:col2] for r in range(row1, row2)]

试试看：

a = Array([['a', 'b', 'c', 'd'],   ['e', 'f', 'g', 'h'],   ['i', 'j', 'k', 'l'],   ['m', 'n', 'o', 'p'],   ['q', 'r', 's', 't'],   ['u', 'v', 'w', 'x']])print(a[1:5, 1:3])

官方文档对__getitem__的解释：

简而言之，其主要用来获取self[key]的值。

我在这里为了突出问题解决，只列出了关键代码，异常判断、边界检查、条件限制，甚至其他一些特殊方法如__setitem__ 、 __delitem__与__len__等代码，需视实际情况添加。

当然，也有其他处理方式，如以下所示代码，但不同方法无疑给了我各种场景下的多种选项。

a = [['a', 'b', 'c', 'd'],  ['e', 'f', 'g', 'h'],  ['i', 'j', 'k', 'l'],  ['m', 'n', 'o', 'p'],  ['q', 'r', 's', 't'],  ['u', 'v', 'w', 'x']]sl = lambda row1, row2, col1, col2, lst: /  [lst[r][col1:col2] for r in range(row1, row2)]sl(1, 5, 1, 3, a)