欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python itertools的使用

程序员文章站 2024-02-12 13:17:58
...

简介

Python的内建模块itertools提供了非常有用的用于操作迭代对象的函数。

无限迭代器

  • count
    count()会创建一个无限的迭代器,所以上述代码会打印出自然数序列,根本停不下来,只能按Ctrl+C退出。
import itertools
natuals = itertools.count(1)
for n in natuals:
    print n
...
1
2
3
...
  • cycle
    cycle()会把传入的一个序列无限重复下去
import itertools
cs = itertools.cycle('ABC') # 注意字符串也是序列的一种
for c in cs:
...     print c
...
'A'
'B'
'C'
'A'
'B'
'C'
...
  • repeat
    repeat()负责把一个元素无限重复下去,不过如果提供第二个参数就可以限定重复次数
ns = itertools.repeat('A', 10)
for n in ns:
...     print n
...
打印10次'A'
  • takewhile
    chain()可以把一组迭代对象串联起来,形成一个更大的迭代器
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 10, natuals)
for n in ns:
...     print n
...
打印出1到10

以上来自廖雪峰网站

chain方法

chain()可以把一组迭代对象串联起来,形成一个更大的迭代器:

这个方法自己平时用的比较多,下面讲一下关于chain的实例

  • 实例 1
for c in itertools.chain('ABC', 'XYZ'):
    print c
# 迭代效果:'A' 'B' 'C' 'X' 'Y' 'Z
  • 实例 2
import itertools
first=[['1','2','3','4','4'],['2','4','2']]
print(*first)
x=['abc','def']
print(*x)
for i in itertools.chain(*first):
    print(i)

输出为

['1', '2', '3', '4', '4'] ['2', '4', '2'] =>这里是两个元素
abc def =>这里是两个元素
1
2
3
4
4
2
4
2

*放在一个可迭代对象前面可以将对象拆分成多个单元素,比如first=[['1','2','3','4','4'],['2','4','2']],如果把*放在first前面,就会拆成['1','2','3','4','4']['2','4','2']这两个元素,然后在使用chain方法进行迭代

  • 实例3
from itertools import chain
s1=['i','l','o','v','e']
s2=['y','o','u']
s=[s1,s2]

all_letters=chain(*s)
print(all_letters,type(all_letters))
print(list(all_letters))

输出为

<itertools.chain object at 0x00000000026DA400> <class 'itertools.chain'>
['i', 'l', 'o', 'v', 'e', 'y', 'o', 'u']