2015-07-22 42 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保留了输入的顺序,并且由于我们传递了一个非递减序列,所以我们只得到非递减的元组。

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