微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐