2013-06-18 1484 views
10

我知道我可以使用itertools.permutation來獲取大小爲r的所有排列組合。 但是,對於itertools.permutation([1,2,3,4],3)它將返回(1,2,3)以及(1,3,2)Python - 獲取列表的所有排列不重複

  1. 我想過濾那些重複(即獲得組合)

  2. 有沒有一種簡單的方法來獲得(所有長度)的所有排列?

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

+1

你在找'組合嗎? –

回答

19

使用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)] 
+0

什麼?一個5年upvotes的問題(我將獲得第六名)已經提到'itertools'作爲* itertool *並且沒有人設法修復這個錯字? –

4

您需要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)] 
5

這聽起來像你實際上是在尋找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)]