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

Question

在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]],等一切算相同,所以只有其中一人应在整个迭代显示.

Answer 1

你会想要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]]

正如@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]]

编辑

在仔细阅读你的问题时,不清楚你是否想要n/2像我所示的所有排列,或者你想要一个lits列表,其中每个元素都是排列的两个"一半"的另一个列表.

为此,您可以执行以下操作(包含@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和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]]]