如何使用 itertools-Python 找到最小长度为 N 的元素组合

Question

from itertools import combinations
a = [1,2,3]
combinations(a,2) #will give me ((1,2),(1,3),(2,3))
combinations(a,3) #will give me ((1,2,3),)

但是如果我想要数组中不同长度的结果怎么办 例如

我想找到长度大于或等于 2 的给定数组 [1,2,3] 的所有组合 所以结果应该是((1,2),(1,3),(2,3),(1,2,3))

类似c = combinations(a,>=2) 我尝试使用 lambda 但它不起作用

c = combinations(a,lambda x: x for x in [2,3])

同时列出综合c = combinations(a,[x for x in [2,3]])

我知道我可以使用一个简单的循环，然后找出 diff 长度的组合。

for l in [2,3]:
    combinations(a,l)

但是有什么pythonic方法可以做到这一点吗？

Answer 1

你可以结合combinations和chain.from_iterable：

>>> from itertools import chain, combinations
>>> a = [1,2,3]
>>> n = 2
>>> cc = chain.from_iterable(combinations(a, i) for i in range(n, len(a)+1))
>>> list(cc)
[(1, 2), (1, 3), (2, 3), (1, 2, 3)]

chain.from_iterable 这里将生成器表达式(combinations(a, i) for i in range(n, len(a)+1)) 生成的内容展平。否则你会得到类似的东西

>>> [list(combinations(a,i)) for i in range(n, len(a)+1)]
[[(1, 2), (1, 3), (2, 3)], [(1, 2, 3)]]

这很好，但与您要查找的格式不完全一致。