举个combinations函数的例子
时间: 2023-09-12 12:07:58 浏览: 77
组合模式示例(combination-demo)
好的,下面是一个使用Python内置的combinations函数的例子:
假设我们有一个数组[1, 2, 3, 4],我们想从中选择两个数的所有可能组合。
代码如下:
```python
from itertools import combinations
arr = [1, 2, 3, 4]
comb = combinations(arr, 2)
for i in comb:
print(i)
```
输出结果如下:
```
(1, 2)
(1, 3)
(1, 4)
(2, 3)
(2, 4)
(3, 4)
```
以上代码使用了Python标准库中的itertools模块中的combinations函数,该函数可以返回一个可迭代对象,其中包含了给定数组中选取指定数量元素的所有可能组合。在上面的例子中,我们给定的数组为[1, 2, 3, 4],并指定选取2个元素,所以我们得到了所有2个元素的可能组合。
阅读全文