Python itertools模块中的combinations() 函数用法

调用方法

itertools.combinations(iterable, r)
1

各个参数意义：
iterable：输入数据，数据应该是可迭代的。
r：子序列的长度
返回值：从输入的可迭代数据 iterable 中返回 r 长度的子序列。

示例1

from itertools import combinations

results = combinations("ABCD", 2)
for result in results:
    print(result)
"""
result:
('A', 'B')
('A', 'C')
('A', 'D')
('B', 'C')
('B', 'D')
('C', 'D')
"""
1
2
3
4
5
6
7
8
9
10
11
12
13
14

从输出的结果可以看出，combinations() 函数返回的是长度为 2 的 "ABCD" 字符串的子序列。

示例2

from itertools import combinations

results = combinations(range(4), 3)
for result in results:
    print(result)
"""
result:
(0, 1, 2)
(0, 1, 3)
(0, 2, 3)
(1, 2, 3)
"""
1
2
3
4
5
6
7
8
9
10
11
12

range(4) 中包含的元素为 [0,1,2,3]，然后返回它的长度为 3 的子序列，结果如上面显示。

如果大家觉得有用，就点个赞让更多的人看到吧~