python中itertools模块使用小结

itertools 是python的迭代器模块，itertools提供的生成迭代器的函数，相当高效且节省内存。使用这些工具，你将能够创建自己定制的迭代器用于高效率的循环。接下来通过本文给大家介绍python中itertools模块使用，感兴趣的朋友一起看看吧

Python 内置的 itertools 模块包含了一系列用来产生不同类型迭代器的函数或类，这些函数的返回都是一个迭代器，我们可以通过 for 循环来遍历取值，也可以使用 next() 来取值。

itertools 模块提供的迭代器函数有以下几种类型：

无限迭代器：生成一个无限序列，比如自然数序列 1, 2, 3, 4, ...；
有限迭代器：接收一个或多个序列（sequence）作为参数，进行组合、分组和过滤等；
组合生成器：序列的排列、组合，求序列的笛卡儿积等。

itertools

高效循环下创建循环器的标准库

Infinite itertools，无限迭代器

itertools.count(start=0, step=10)

默认返回一个从0开始，依次+10的自然数迭代器，如果你不停止，它会一直运行下去。
可以用start指定开始的位置，step是步长。

 import itertools c = itertools.count(start=0, step=1) for i in c: print(i) # 0 # 10 # 20 # 30 # 40 # 50 # ...

itertools.cycle(iterable)

传入一个可迭代对象，然后无限循环迭代。

 import itertools # itertools.count() l = [1,2,3,4,5] c = itertools.cycle(l) for i in c: print(i) # 1 # 2 # 3 # 4 # 5 # 1 # 2 # 3 # 4 # 5 # ...

itertools.repeat(p_object, times=None)

重复迭代一个对象p_object，如果不指定times，则会迭代无数次，也可以通过times参数指定迭代的次数。

 import itertools # itertools.count() l = [1,2,3,4,5] c = itertools.repeat(l, 5) for i in c: print(i) # [1, 2, 3, 4, 5] # [1, 2, 3, 4, 5] # [1, 2, 3, 4, 5] # [1, 2, 3, 4, 5] # [1, 2, 3, 4, 5]

Iterators terminating on the shortest input sequence

itertools.accumulate(iterable, func)

返回序列的累计值或者其他二进制函数。

 import itertools # itertools.count() l = [1,2,3,4,5] c = itertools.accumulate(l) print(c) for i in c: print(i) # 1 # 3 # 6 # 10 # 15

accumulate()仍然返回的是一个迭代器，传一个list，在for循环中遍历打印的时候发现，它做了累加操作。(迭代第一个数，就是前一个数的和，迭代到第二个数时，就是前两个数的和，以此类推)

并且做递加操作时支持：list, tuple, str, set, dict

传入的是dict对象，那么会累加迭代dict的key：

 import itertools # itertools.count() d = {'a': 1, 'b': 2, 'c': 3} c = itertools.accumulate(d) print(c) for i in c: print(i) #  # a # ab # abc

itertools.chain(*iterables)

chain()方法中的参数可以传入多个序列，而且只要是序列即可，不限定序列的数据类型。

如：迭代list, tuple, str三个序列

 import itertools l = [1, 2, 3, 4, 5] t = (1, 2, 3, 4, 5) s = 'abcdefg' c = itertools.chain(l, t, s) print(c) for i in c: print(i) #  # 1 # 2 # 3 # 4 # 5 # 1 # 2 # 3 # 4 # 5 # a # b # c # d # e # f # g

itertools 笛卡尔积

 import itertools d = [ [{"a": 1}, {"b": 2}], [{"c": 3}, {"d": 4}, {"e": 5}], [{"f": 6}, {"g": 7}] ] print(*d) for i in itertools.product(*d): print(i) # [{'a': 1}, {'b': 2}] [{'c': 3}, {'d': 4}, {'e': 5}] [{'f': 6}, {'g': 7}] # ({'a': 1}, {'c': 3}, {'f': 6}) # ({'a': 1}, {'c': 3}, {'g': 7}) # ({'a': 1}, {'d': 4}, {'f': 6}) # ({'a': 1}, {'d': 4}, {'g': 7}) # ({'a': 1}, {'e': 5}, {'f': 6}) # ({'a': 1}, {'e': 5}, {'g': 7}) # ({'b': 2}, {'c': 3}, {'f': 6}) # ({'b': 2}, {'c': 3}, {'g': 7}) # ({'b': 2}, {'d': 4}, {'f': 6}) # ({'b': 2}, {'d': 4}, {'g': 7}) # ({'b': 2}, {'e': 5}, {'f': 6}) # ({'b': 2}, {'e': 5}, {'g': 7})

到此这篇关于python中的itertools模块简单使用的文章就介绍到这了,更多相关python itertools模块使用内容请搜索0133技术站以前的文章或继续浏览下面的相关文章希望大家以后多多支持0133技术站！

以上就是python中itertools模块使用小结的详细内容，更多请关注0133技术站其它相关文章！