基於一個關鍵屬性的python嵌套列表中的組合

Question

我不知道如何描述這個。我將展示一個例子： 這是我的輸入列表

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

這是輸出我想：

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

即我想要一個包含取2內列出的所有組合在時間列表，基於內部列表的第一個字段。

我試圖itertools是這樣的：

perm_list = itertools.combinations(task_list, 10) 

但它是所有顯示此：

itertools.combinations object at 0x0120B060 

Answer 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 

如果您不需要存儲結果，請不要構造列表。