Python 生成所有非递减序列

Question

我无法找到以 Pythonic 方式执行此操作的方法。我想我可以以某种方式使用 itertools，因为我以前做过类似的事情，但不记得我做了什么。

我正在尝试生成所有长度为 L 的非递减列表，其中每个元素的值都介于 1 和 N 之间。例如，如果 L=3 且 N=3，则 [1,1,1]， [1,1,2],[1,1,3],[1,2,2],[1,2,3], 等等

Answer 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 保留了输入的顺序，并且由于我们传入的是非递减序列，所以我们只会得到非递减的元组。

（如果您确实需要列表而不是元组，则很容易转换为列表。）

Python 生成所有非递减序列

Python generating all nondecreasing sequences

python

list

itertools