对数组进行排列组合 python
时间: 2024-02-02 08:38:36 浏览: 64
python-leetcode面试题解之第373题查找和最小的K对数字.zip
可以使用itertools库中的permutations和combinations函数来对数组进行排列组合。
例如,对于一个数组arr=[1,2,3],可以使用permutations函数来获取其所有的排列:
```python
import itertools
arr = [1,2,3]
permutations = list(itertools.permutations(arr))
print(permutations)
```
输出结果为:
```
[(1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)]
```
同样地,可以使用combinations函数来获取其所有的组合:
```python
import itertools
arr = [1,2,3]
combinations = []
for i in range(1, len(arr)+1):
combinations += list(itertools.combinations(arr, i))
print(combinations)
```
输出结果为:
```
[(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]
```
阅读全文