簡體   English   中英

Python生成所有非遞減序列

[英]Python generating all nondecreasing sequences

我很難找到一種以Python方式進行此操作的方法。 我想我可以以某種方式使用itertools,因為我之前做過類似的事情,但不記得自己做了什么。

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

您可以使用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保留了輸入的順序,並且由於我們要傳入一個不遞減的序列,所以我們只會取出不遞減的元組。

(如果您確實需要列表而不是元組,則很容易轉換為列表。)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM