itertools：Python3迭代库（持续更新ing...）

诸神缄默不语-个人CSDN博文目录

itertools库是Python3生成可迭代对象的库。
官方文档：itertools — Functions creating iterators for efficient looping — Python 3.10.7 documentation

1. combinations(iterable, r)

返回的可迭代对象中每一个元素是iterable元素的一个组合（按iterable的顺序生成），长度为r。
没有iterable中元素和本元素的组合（没有自环），不包含列表中的重复元素。

示例：

from itertools import combinations
a = ['h', 'y', 'k', 'q', 's']
for i in combinations(a, 2):
    print(i)
1
2
3
4

输出：

('h', 'y')
('h', 'k')
('h', 'q')
('h', 's')
('y', 'k')
('y', 'q')
('y', 's')
('k', 'q')
('k', 's')
('q', 's')
1
2
3
4
5
6
7
8
9
10

2. accumulate(iterable[, func, *, initial=None])

最简单的用法：计算前缀和（直接返回值是一个迭代器）

从第一个数字开始：

import itertools

example_list=[1 for _ in range(10)]
print(list(itertools.accumulate(example_list)))
1
2
3
4

输出：[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

从0开始：

import itertools

example_list=[1 for _ in range(10)]
print(list(itertools.accumulate(example_list,initial=0)))
1
2
3
4

输出：[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

本文撰写过程中参考的网络资料

1. Python中itertools.combinations()的使用_hyk今天写算法了吗的博客-CSDN博客_itertools.combination
2. 还没看完
   1. python3迭代器iter()函数和itertools模块_Ranger L的博客-CSDN博客