N 选择列表的 N/2 个子列表

Python中有没有一种有效的方法来获取大小列表的所有分区n分成两个大小子集n/2？我想获得一些迭代构造，以便每次迭代提供原始列表的两个不重叠的子集，每个子​​集都有大小n/2.

例如：

A = [1,2,3,4,5,6]    # here n = 6
# some iterative construct
    # in each iteration, a pair of subsets of size n/2
    # subsets = [[1,3,4], [2,5,6]] for example for one of the iterations
    # subsets = [[1,2,5],[3,4,6]] a different iteration example

子集应该是不重叠的，例如[[1,2,3], [4,5,6]]是有效的，但是[[1,2,3], [3,4,5]]不是。两个子集的顺序并不重要，例如[[1,2,3], [4,5,6]]不算作不同于[[4,5,6], [1,2,3]]因此，迭代中只应出现这两者之一。每个子集中的顺序也并不重要，所以[[1,2,3], [4,5,6]], [[1,3,2], [4,5,6]], [[3,2,1], [6,5,4]]等都算作相同的，因此在整个迭代中只应显示其中一个。

你会想要使用itertools.combinations去做这个。输入是您要从中选择项目的列表，第二个是要选择的项目数。

result = [list(item) for item in itertools.combinations(input, len(input) // 2)]

对于输入[1,2,3,4]这产生

[[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4]]

As @ShadowRanger 指出，如果顺序在您的列表中很重要并且您想要所有排列，您将需要替换itertools.permutations进入溶液。

result = [list(item) for item in itertools.permutations(input, len(input) // 2)]
# [[1, 2], [1, 3], [1, 4], [2, 1], [2, 3], [2, 4], [3, 1], [3, 2], [3, 4], [4, 1], [4, 2], [4, 3]]

Edit

仔细阅读您的问题后，不清楚您是否想要所有n/2像我已经展示的排列或者你想要一个列表列表，其中每个元素还another排列的两个“一半”的列表。

为了实现这一点，您可以执行以下操作（结合一些索引帮助来自@Blckknght)

result = [[list(item[::2]), list(item[1::2])] for item in itertools.permutations(input)]

在这种情况下，输出[1,2,3,4]将会

[[[1, 3], [2, 4]], [[1, 4], [2, 3]], [[1, 2], [3, 4]], [[1, 4], [3, 2]], [[1, 2], [4, 3]], [[1, 3], [4, 2]], [[2, 3], [1, 4]], [[2, 4], [1, 3]], [[2, 1], [3, 4]], [[2, 4], [3, 1]], [[2, 1], [4, 3]], [[2, 3], [4, 1]], [[3, 2], [1, 4]], [[3, 4], [1, 2]], [[3, 1], [2, 4]], [[3, 4], [2, 1]], [[3, 1], [4, 2]], [[3, 2], [4, 1]], [[4, 2], [1, 3]], [[4, 3], [1, 2]], [[4, 1], [2, 3]], [[4, 3], [2, 1]], [[4, 1], [3, 2]], [[4, 2], [3, 1]]]

Edit2

由于顺序并不重要，但您想要一种类似于上一种方法（列表的列表的列表）的方法，因此由于数组切片，最后一种方法有点棘手。一种替代方法是使用set and frozenset构造初始信息（而不是列表），因为在set检查相等性时顺序并不重要。这将自动允许我们删除重复项。然后，如果您愿意，我们可以添加一个额外的步骤来转换回列表。

from itertools import permutations
tmp = set([frozenset([frozenset(k[::2]),frozenset(k[1::2])]) for k in permutations(input)]) 
result = [[list(el) for el in item] for item in tmp];

这将产生

[[[1, 2], [3, 4]], [[2, 3], [1, 4]], [[1, 3], [2, 4]]]