2

我不知道如何描述这一点。我将展示一个示例:这是我的输入列表

[[0,25], [1,54], [2,76], [3,13], [4,79]]

这是我想要的输出:

[[[0,25], [1,54]], [[0,25], [1,54]], [[0,25], [2,76]], ...]

即我想要一个包含所有内部列表组合的列表,一次取2个,基于内部列表的第一个字段。

我试过这样的迭代工具:

perm_list = itertools.combinations(task_list, 10)

但它所做的只是展示了这一点:

itertools.combinations object at 0x0120B060
4

1 回答 1

1
>>> from itertools import combinations
>>> nums = [[0,25], [1,54], [2,76], [3,13], [4,79]]
>>> list(combinations(nums, r=2))
[([0, 25], [1, 54]), ([0, 25], [2, 76]), ([0, 25], [3, 13]), ([0, 25], [4, 79]), ([1, 54], [2, 76]), ([1, 54], [3, 13]), ([1, 54], [4, 79]), ([2, 76], [3, 13]), ([2, 76], [4, 79]), ([3, 13], [4, 79])]

顾名思义,itertools.combinations返回一个迭代器(懒惰地产生组合),您必须使用带有list(...)构造函数的迭代器来获取您的列表。但通常迭代器更可取,因为您可能只需要迭代组合。

for a, b in combinations(nums, r=2):
    pass

如果您不需要存储结果,请不要构建列表。

于 2013-06-13T10:17:35.383 回答