1.无穷的迭代器

1.1 count(start,[step])

count()接受两个参数

  • start:循环开始的数字
  • step:循环中的间隔
    1. from itertools import count
    2. """
    3. 无穷的迭代器 count()
    4. """
    5. c = count(0, 2)
    6. v = next(c)
    7. while v < 10:
    8.   v = next(c)
    9.   print(v, end=',')
    10. • 1
    11. • 2
    12. • 3
    13. • 4
    14. • 5
    15. • 6
    16. • 7
    17. • 8
    18. • 9
    19. • 10

    1.2 cycle()

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

  1. """
  2. 无穷迭代器 cycle()
  3. """
  4. from itertools import cycle
  5. c = cycle('ABCD')
  6. for i in range(10):
  7.     print(next(c), end=',')
  8. • 1
  9. • 2
  10. • 3
  11. • 4
  12. • 5
  13. • 6
  14. • 7
  15. • 8

1.3 repeat(elem,[n])

重复迭代elem，n次

  1. """
  2. 无穷迭代器 repeat()
  3. """
  4. from itertools import repeat
  5. r = repeat(1, 3)
  6. for i in range(3):
  7.     print(next(r), end=',')
  8. • 1
  9. • 2
  10. • 3
  11. • 4
  12. • 5
  13. • 6
  14. • 7
  15. • 8
  16. • 9

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=',')
• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11

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)
• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8

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)
• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9

    3.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])))
• 1
• 2
• 3
• 4
• 5
• 6

3.5 dropwhile(pred,seq)

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

"""
迭代器 dropwhile()
"""
from itertools import dropwhile
l = [1, 7, 6, 3, 8, 2, 10]
print(list(dropwhile(lambda x: x < 3, l)))
• 1
• 2
• 3
• 4
• 5
• 6
• 7

3.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))
• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9

    3.7 islice

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

from itertools import islice
print(list(islice('ABCDEFG', 2,3, None)))
• 1
• 2

3.8 zip_longest

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

  • zip结束取决于里面最短的迭代对象
  • zip_longest结束取决于里面最长的迭代对象
    from itertools import zip_longest
for 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)
• 1
• 2
• 3
• 4
• 5
• 6

排列组合迭代器

3.1 product

相当于 嵌套的for

“””
排列组合迭代器 product 嵌套的for
“””
from itertools import product
for 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 permutations
print(list(permutations('123')))
• 1
• 2

3.3 combinations(p,r)

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

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

3.4 combinations_with_replacement()

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

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