Python - 獲取列表的所有組合

Question

我知道我可以使用itertools.permutation來獲得大小為r的所有排列。 但是，對於itertools.permutation([1,2,3,4],3) ，它將返回(1,2,3)以及(1,3,2) 。

1. 我想過濾那些重復（即獲得組合）

2. 是否有一種簡單的方法來獲得所有排列（所有長度）？

3. 如何將itertools.permutation()結果轉換為常規列表？

Answer 1

使用itertools.combinations和一個簡單的循環來獲得所有大小的組合。

combinations返回一個迭代器，所以你要將它傳遞給list()以查看它的內容（或使用它）。

>>> from itertools import combinations
>>> lis = [1, 2, 3, 4]
for i in xrange(1, len(lis) + 1):  #  xrange will return the values 1,2,3,4 in this loop
    print list(combinations(lis, i))
...     
[(1,), (2,), (3,), (4,)]
[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]
[(1,2,3,4)]

Answer 2

聽起來你實際上在尋找itertools.combinations() ：

>>> from itertools import combinations
>>> list(combinations([1, 2, 3, 4], 3))
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]

此示例還顯示了如何將結果轉換為常規列表，只需將其傳遞給內置list()函數。

要獲得每個長度的組合，您可以使用如下所示的循環：

>>> data = [1, 2, 3, 4]
>>> for i in range(1, len(data)+1):
...     print list(combinations(data, i))
... 
[(1,), (2,), (3,), (4,)]
[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]
[(1, 2, 3, 4)]

或者，要將結果作為嵌套列表，您可以使用列表推導：

>>> [list(combinations(data, i)) for i in range(1, len(data)+1)]
[[(1,), (2,), (3,), (4,)], [(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)], [(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)], [(1, 2, 3, 4)]]

對於平面列表而不是嵌套：

>>> [c for i in range(1, len(data)+1) for c in combinations(data, i)]
[(1,), (2,), (3,), (4,), (1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4), (1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4), (1, 2, 3, 4)]

Answer 3

你需要itertools.combinations() 。 要獲得常規列表，只需使用list()工廠函數。

>>> from itertools import combinations
>>> list(combinations([1, 2, 3, 4], 3))
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]