网络编程
位置:首页>> 网络编程>> Python编程>> Python中二维列表如何获取子区域元素的组成

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

作者:瓜园耕读  发布时间:2021-02-16 02:41:01 

标签:python,二维列表,元素

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

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

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

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

但有时候我只想用一下这个功能,但又不想引入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])

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

官方文档对__getitem__的解释:

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

简而言之,其主要用来获取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)

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

来源:http://www.2gua.info/post/72

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com