Itertools in python

Itertools in python

  • 定义
  1. Itertools:Functions creating iterators for efficient looping(是一系列函数,用来创建迭代器,是为了更有效的循环。
  2. 迭代器:可以看成是一个for循环的封装。有无限循环和有限循环
    举个例子:
// 无限循环迭代器cycle
cycle('ABCD') --> A B C D A B C D ...
  1. 对于Itertool functions,它是用来创建和返回一个迭代器。
# itertools.chain(*iterables) 创建一个迭代器
def chain(*iterables):
    # chain('ABC', 'DEF') --> A B C D E F
    for it in iterables:
        for element in it:
            yield element
# classmethod chain.from_iterable(iterable) 类似chain()
def from_iterable(iterables):
    # chain.from_iterable(['ABC', 'DEF']) --> A B C D E F
    for it in iterables:
        for element in it:
       	# yield 相当于return, 只是它返回的是一种iterator
            yield element
  • 通俗来说,这两个迭代器可以把它看作是把这些列表打碎了给它的元素提取出来。也可以看作是flatten扁平化的一个过程。
    举例:
# output: [{'a': 0}, {'b': 1}, {'c': 2}, {'d': 3}]
ite = itertools.chain.from_iterable(
	[
		[{'a':0}, {'b':1}], 
		[{'c':2}], 
		[{'d':3}]
	]
)
print(list(ite)) 

  • 总结
    所以当结构嵌套多层的时候,可以用这个迭代器的方法高效遍历。

  • 可以继续学习的点:
    迭代器结合map,用法更高效。

参考文献

https://docs.python.org/zh-cn/3/library/itertools.html#itertools.chain

相关文章

在前一篇博客中我们介绍了加侧旋的乒乓球弧圈技术的模拟,本...
在近期conda的版本更新中,有可能会删除路径下的_sysconfigd...
本文主要展示了一些lambda表达式的使用示例,通过这些示例,...
本文通过对比Jax和Numpy计算Normalized Hamming Distance的过...
我们知道GPU加速在可并行化程度比较高的算法中,能够发挥出比...
Numpy这个库在Python编程中非常的常用,不仅在性能上补足了P...