简介在Python/ target=_blank class=infotextkey>Python中 , 迭代器可以帮助你编写更多Pythonic的代码,并在处理长序列时提高效率 。内置的itertools
模块提供了几个有用的函数来创建迭代器 。
【itertools
】:https://docs.python.org/3/library/itertools.html
当你只需要遍历迭代器、检索序列中的元素并对其进行处理,而无需将它们存储在内存中时,这些函数尤其有用 。今天本文将学习如何使用以下四个itertools
过滤函数:
- filterfalse
- takewhile
- dropwhile
- islice
文章插图
在开始之前:关于代码示例的说明在本教程中:
我们将讨论的所有四个函数都返回了迭代器 。为了清楚起见 , 本文将使用简单的序列 , 并使用
list()
获取包含迭代器返回的所有元素的列表 。但在处理长序列时,除非必要,否则请不要这样做,因为这样做会失去迭代器带来的内存节省 。对于简单的谓词函数,也可以使用
lambdas
。但为了提高可读性,本文将定义常规函数并将它们用作谓词 。1. filterfalse如果你在Python中编程已经有一段时间了,可能已经使用过内置的
filter
函数,语法如下:filter(pred,seq)# pred:谓词函数# seq:任何有效的Python可迭代对象
filter
函数返回一个迭代器 , 该迭代器返回谓词函数返回True
的序列中的元素 。示例如下:
nums = list(range(1,11)) #[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]def is_even(n): return n % 2 == 0
在这里 , nums
列表和is_even
函数分别是序列和谓词 。要获取
nums
中所有偶数的列表,需要使用如下所示的filter
:nums_even = filter(is_even, nums)print(list(nums_even))
Output >>> [2, 4, 6, 8, 10]
现在跟随本文来学习一下filterfalse
。本文将从itertools
模块中导入filterfalse
函数(以及本文将要讨论的所有其他函数) 。正如其名称所示,
filterfalse
执行与filter
函数相反的操作 。它返回一个迭代器,该迭代器返回谓词返回False
的元素 。以下是使用filterfalse
函数的语法:from itertools import filterfalsefilterfalse(pred,seq)
is_even
函数对于nums
中的所有奇数返回False
。因此,使用filterfalse
得到的nums_odd
列表是nums
中所有奇数的列表:from itertools import filterfalsenums_odd = filterfalse(is_even, nums)print(list(nums_odd))
Output >>> [1, 3, 5, 7, 9]
2. takewhile使用takewhile
函数的语法如下:from itertools import takewhiletakewhile(pred,seq)
takewhile
函数返回了一个迭代器,只要谓词函数返回True
,它就会返回元素 。当谓词函数第一次返回False
时 , 它就停止返回元素 。对于长度为n的序列,如果
seq[k]
是第一个使谓词函数返回False
的元素,则迭代器会返回seq[0]
、seq[1]
、...、seq[k-1]
。考虑以下的
nums
列表和谓词函数is_less_than_5
。本文使用takewhile
函数 , 如下所示:from itertools import takewhiledef is_less_than_5(n): return n < 5nums = [1, 3, 5, 2, 4, 6]filtered_nums_1 = takewhile(is_less_than_5, nums)print(list(filtered_nums_1))
在这里,谓词is_less_than_5
对于数字5
首次返回False
:Output >>> [1, 3]
3. dropwhile从功能上讲,dropwhile
函数的作用与takewhile
函数相反 。以下是如何使用
dropwhile
函数的示例:from itertools import dropwhiledropwhile(pred,seq)
dropwhile
函数返回一个迭代器,只要谓词为True
,该迭代器就会持续删除元素 。也就是说,迭代器在谓词第一次返回
推荐阅读
- 五个帮助初学者提高编码技术的简单方法
- AI绘画 | 纯净美丽的白色薄纱少女
- “维护者都快累死了!”Linux 宣布:LTS 版本的维护期,将从 6 年变回 2 年
- 深度学习的关键特征是什么
- 解决梯度消失问题的神经网络结构及其广泛应用
- AI绘画 | 黑发及腰的魅力少女
- 深度网络的退化问题
- 数字人点燃杭州亚运会主火炬!亚运史首个!到底是如何实现的?
- 六个Spring Boot处理异常的小技巧
- C++11中auto关键字的深入理解与应用