python中的itertools的使用详解

今天了解了下python中内置模块itertools的使用，熟悉下，看能不能以后少写几个for,嘿嘿😁。

1.无穷的迭代器

1.1 count(start,[step])

count()接受两个参数

• start:循环开始的数字
• step:循环中的间隔

from itertools import count

"""
无穷的迭代器 count()
"""
c = count(0,2)
v = next(c)
while v < 10:
  v = next(c)
  print(v,end=',')

1.2 cycle()

cycle就是一while True，无限循环里面的数字。

"""
无穷迭代器 cycle()
"""
from itertools import cycle

c = cycle('ABCD')
for i in range(10):
  print(next(c),')

1.3 repeat(elem,[n])

重复迭代elem，n次

"""
无穷迭代器 repeat()
"""
from itertools import repeat

r = repeat(1,3)
for i in range(3):
  print(next(r),')

2. 迭代器

2.1 accumulate(p,[func])

使用func的函数对迭代对象p进行累积。

"""
迭代器 accumulate()
"""
from itertools import accumulate

test_list = [i for i in range(1,11)]
for i in accumulate(test_list): # 默认是operator.add
  print(i,')
print()
for i in accumulate(test_list,lambda x,y: x * y): # operator.mul
  print(i,')

2.2 chain()

chain()中可以放多个迭代对象，然后一一迭代出来。

"""
迭代器 chain()
"""
from itertools import chain

ch = chain([1,2,3],{4: 4,5: 5},{6,7,8},(9,),[10,[11,12]])
for i in ch:
  print(i)

2.3 chain.from_iterable()

跟chain不同的地方在于:

• chain: 可以接受多个迭代对象
• chain.from_iterable():可以接受一个可以产生迭代对象的迭代器

"""
迭代器 chain.from_iterable()
"""
def gen_iterables():
  for i in range(10):
    yield range(i)

for i in chain.from_iterable(gen_iterables()):
  print(i)

2.4 compress(data,selectors)

这是就是看下这个就知道了s是selectors中的元素。
(d[0] if s[0]),(d[1] if s[1]),...

"""
迭代器 compress
"""
from itertools import compress

print(list(compress(['A','B','C','D'],[0,1,1])))

2.5 dropwhile(pred,seq)

循环开始的条件是，直到遇到第一次不满足pred条件的情况，才开始遍历。

"""
迭代器 dropwhile()
"""
from itertools import dropwhile

l = [1,6,3,8,10]
print(list(dropwhile(lambda x: x < 3,l)))

2.6 groupby

这个感觉挺有意思的，有点像sql中的group_by。可以对字符串，列表等进行分组。

返回键和，组里的内容

from itertools import groupby

# 对字符串进行分组
for k,g in groupby('22222234567'):
  print(k,list(g))
d = {1: 1,2: 2,3: 2}
# 按照字典value来进行分组
for k,g in groupby(d,lambda x: d.get(x)):
  print(k,list(g))

2.7 islice
这个就是对迭代对象进行切割，不支持负数，有点像range(1,10,2)这种

from itertools import islice
print(list(islice('ABCDEFG',None)))

2.8 zip_longest

这个和zip很像，不同地方在于:

• zip结束取决于里面最短的迭代对象
• zip_longest结束取决于里面最长的迭代对象

from itertools import zip_longest

for x,y in zip_longest([1,[1,2]):
  print(x,y)
for x,y in zip([1,y)

排列组合迭代器

3.1 product

相当于 嵌套的for

“”"
排列组合迭代器 product 嵌套的for
“”"
from itertools import product
for i,j in product([1,[4,5]):
print(i,j

3.2 permutations

全排列，比如输出123的全部情况。(1,3),(1,2)…

from itertools import permutations
print(list(permutations('123')))

3.3 combinations(p,r)

从p中找出所有长度为r的排列情况… 有顺序

from itertools import combinations
print(list(combinations([1,2)))

3.4 combinations_with_replacement()

从p中找出所有长度为r的排列情况，有顺序，但包括自身就是会重复的意思。

• combinations_with_replacement(‘ABCD',2)
• AA AB AC AD BB BC BD CC CD DD

了解是了解了，就是用的时候不知道能不能想起来…

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!