python 求组合

在 Python 中，可以使用 itertools 模块的 combinations 函数来求组合。

语法格式如下：

itertools.combinations(iterable, r)

• iterable: 要获取组合的元素列表
• r: 获取组合的元素个数

返回一个迭代器，它返回的是 iterable 中的元素的长度为 r 的所有组合。

例如：

import itertools

elements =[1, 2, 3, 4]
combinations = list(itertools.combinations(elements, 2))
print(combinations)

输出：

[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]

这里展示了对元素1,2,3,4的两两组合方式

也可以通过for循环来实现这个过程

elements =[1, 2, 3, 4]
for i in range(len(elements)):
    for j in range(i+1,len(elements)):
        print(elements[i],elements[j])

在这里会按顺序输出两两组合，输出结果与上面itertools.combinations(elements, 2)是一致的.