Python生成所有非递减序列

时间:2015-07-22 00:52:15

标签: python list itertools

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

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

1 个答案:

答案 0 :(得分: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保留了输入的顺序,并且由于我们正在传递非递减序列,所以我们只得到非递减的元组。

(如果您真的需要那些而不是元组,那么很容易转换为列表。)