python从n个数里选k个元素组合
时间: 2023-11-18 21:05:18 浏览: 86
可以使用Python内置的itertools模块中的combinations函数来实现从n个数中选k个元素的组合。
具体实现方法如下:
```python
import itertools
n = 5
k = 3
nums = list(range(1, n+1))
combinations = list(itertools.combinations(nums, k))
print(combinations)
```
输出结果为:`[(1, 2, 3), (1, 2, 4), (1, 2, 5), (1, 3, 4), (1, 3, 5), (1, 4, 5), (2, 3, 4), (2, 3, 5), (2, 4, 5), (3, 4, 5)]`,表示从1到5这5个数中选取3个元素的所有组合。
阅读全文