我很难找到一种以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保留了输入的顺序,并且由于我们要传入一个不递减的序列,所以我们只会取出不递减的元组。

(如果确实需要列表而不是元组,则很容易转换为列表。)

10-06 00:53