多组长度数组比较取得n个不重复组合
时间: 2024-02-12 14:04:52 浏览: 54
以下是一个实现n个不重复组合的Python代码示例,其中使用了itertools库中的combinations函数:
```python
import itertools
def get_combinations(arr, n):
res = []
for i in range(1, len(arr)+1):
for c in itertools.combinations(arr, i):
if len(c) == n and c not in res:
res.append(c)
return res
arr1 = [1, 2, 3]
arr2 = [4, 5, 6, 7]
arr3 = [8, 9]
combinations = get_combinations([arr1, arr2, arr3], 3)
print(combinations)
```
这个函数接收两个参数,分别是多组长度数组和需要取得的组合元素个数n。首先遍历每个数组,然后使用itertools.combinations函数获取该数组中所有可能的组合,再判断组合中元素个数是否等于n并且是否已经存在于结果列表中,最终返回所有符合条件的组合。
在上面的示例中,我们传入了三个数组arr1、arr2和arr3,需要取得长度为3的不重复组合。运行结果为:
```
[(1, 2, 4), (1, 2, 5), (1, 2, 6), (1, 2, 7), (1, 3, 4), (1, 3, 5), (1, 3, 6), (1, 3, 7), (2, 3, 4), (2, 3, 5), (2, 3, 6), (2, 3, 7)]
```
可以看到,该函数成功返回了所有长度为3的不重复组合。
阅读全文