將列表分成所有組合

Question

我知道有很多問題類似的帖子，並且已經解決了很多問題。 但是，我無能為力。

我有列表L = [0,1,2,3,4,5]我想將其拆分為一對元組。 例如：

[(0,1,2),(3,4,5)]
[(0,1,3),(2,4,5)]
[(0,1,4),(2,4,5)]
...

每個元組都需要包含原始列表中一半的元素（在此示例中為3之6）。 一個解決方案需要使用3個元素來生成元組的每種組合。

我可以使用以下命令輕松找到列表中所有可能的元組

list(itertools.combinations(L, 3))

[(0, 1, 2), (0, 1, 3), (0, 1, 4), (0, 1, 5), (0, 2, 3), (0, 2, 4), ...]

itertools是否也為此提供任何工作環境？

Answer 1

可能會有更多性能更好的解決方案避免對列表的另一半進行完整的額外迭代，但這應該可以忽略不計：

l = [[x, tuple(y for y in L if y not in x)] for x in combinations(L, 3)]
[[(0, 1, 2), (3, 4, 5)],
 [(0, 1, 3), (2, 4, 5)],
 [(0, 1, 4), (2, 3, 5)],
 [(0, 1, 5), (2, 3, 4)],
 [(0, 2, 3), (1, 4, 5)],
 [(0, 2, 4), (1, 3, 5)],
 [(0, 2, 5), (1, 3, 4)],
 [(0, 3, 4), (1, 2, 5)],
 [(0, 3, 5), (1, 2, 4)],
 [(0, 4, 5), (1, 2, 3)],
 [(1, 2, 3), (0, 4, 5)],
 [(1, 2, 4), (0, 3, 5)],
 [(1, 2, 5), (0, 3, 4)],
 [(1, 3, 4), (0, 2, 5)],
 [(1, 3, 5), (0, 2, 4)],
 [(1, 4, 5), (0, 2, 3)],
 [(2, 3, 4), (0, 1, 5)],
 [(2, 3, 5), (0, 1, 4)],
 [(2, 4, 5), (0, 1, 3)],
 [(3, 4, 5), (0, 1, 2)]]

這取決於原始列表中是否沒有重復項。 否則，您將不得不使用索引。 以下修改使用相同的方法，但是將列表索引用於組合，因此可以處理原始列表中的重復項：

indexes = ((x, (y for y in L if y not in x)) for x in combinations(range(len(L)), 3))
l = [[tuple(L[a] for a in A), tuple(L[b] for b in B)] for A, B in indexes]