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 cyclec = cycle('ABCD')for i in range(10):

print(next(c), end=',')

1.3 repeat(elem,[n])重复迭代elem，n次

\"\"\"

⽆穷迭代器 repeat()\"\"\"

from itertools import repeatr = repeat(1, 3)for i in range(3):

print(next(r), end=',')

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, end=',')print()

for i in accumulate(test_list, lambda x, y: x * y): # operator.mul print(i, end=',')

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, 1])))

2.5 dropwhile(pred,seq)

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

\"\"\"

迭代器 dropwhile()\"\"\"

from itertools import dropwhilel = [1, 7, 6, 3, 8, 2, 10]

print(list(dropwhile(lambda x: x < 3, l)))

2.6 groupby

这个感觉挺有意思的，有点像sql中的group_by。可以对字符串，列表等进⾏分组。返回键和，组⾥的内容

from itertools import groupby

# 对字符串进⾏分组

for k, g in groupby('11111234567'): 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', 2,3, None)))

2.8 zip_longest

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

zip结束取决于⾥⾯最短的迭代对象

zip_longest结束取决于⾥⾯最长的迭代对象

from itertools import zip_longestfor x,y in zip_longest([1,2,3],[1,2]): print(x,y)

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

排列组合迭代器3.1 product相当于 嵌套的for

“”\"

排列组合迭代器 product 嵌套的for“”\"

from itertools import productfor i,j in product([1,2,3],[4,5]):print(i,j

3.2 permutations

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

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

3.3 combinations(p,r)

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

from itertools import combinationsprint(list(combinations([1,2,3],2)))

3.4 combinations_with_replacement()

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

combinations_with_replacement(‘ABCD', 2)AA AB AC AD BB BC BD CC CD DD了解是了解了，就是⽤的时候不知道能不能想起来…

以上就是本⽂的全部内容，希望对⼤家的学习有所帮助，也希望⼤家多多⽀持。