2015-07-22 50 views
1

我無法找到一種方法在Python化的方式來做到這一點的所有非降序列。我認爲我可以以某種方式使用itertools,因爲我之前做過類似的事情,但不記得我做了什麼。Python的產生

我試圖產生長度L的所有非遞減列表,其中每個元素可以取的值在1和N之間。例如,如果L = 3和N = 3,那麼[1,1,1],[ 1,1,2],[1,1,3],[1,2,2],[1,2,3]等

回答

1

可以使用itertools.combinations_with_replacement做到這一點:

>>> L, N = 3,3 
>>> cc = combinations_with_replacement(range(1, N+1), L) 
>>> for c in cc: print(c) 
(1, 1, 1) 
(1, 1, 2) 
(1, 1, 3) 
(1, 2, 2) 
(1, 2, 3) 
(1, 3, 3) 
(2, 2, 2) 
(2, 2, 3) 
(2, 3, 3) 
(3, 3, 3) 

這是有效的,因爲c_w_r保留了輸入的順序,並且由於我們傳遞了一個非遞減序列,所以我們只得到非遞減的元組。

(如果你真的需要這些元組而不是元組,很容易轉換爲列表。)