我怎样才能生成这个？

Question

给定a list,[3, 2, 4]如何,我如何生成以下输出？

[0, 0, 0]
[0, 0, 1]
[0, 0, 2]
[0, 0, 3]
[0, 0, 4]
[0, 1, 0]
[0, 1, 1]
[0, 1, 2]
[0, 1, 3]
[0, 1, 4]
[0, 2, 0]
[0, 2, 1]
[0, 2, 2]
[0, 2, 3]
[0, 2, 4]
[1, 0, 0]
...
[1, 2, 4]
...
[3, 2, 4]

我需要生成每个列表,其中列表的每个元素都是正数,并且不大于相关索引处给出的数字.

我知道这涉及for循环,但我正在寻找可以处理不同大小的列表的最有效方式(可能使用生成器).

Answer 1

itertools.product() 可以做的工作:

from itertools import product
l = [3,2,4]
print(*product(*[range(i+1) for i in l]), sep='\n')

会打印:

(0, 0, 0)
(0, 0, 1)
(0, 0, 2)
(0, 0, 3)
(0, 0, 4)
(0, 1, 0)
(0, 1, 1)
(0, 1, 2)
(0, 1, 3)
(0, 1, 4)
(0, 2, 0)
(0, 2, 1)
(0, 2, 2)
(0, 2, 3)
(0, 2, 4)
(1, 0, 0)
(1, 0, 1)
(1, 0, 2)
...
(3, 2, 3)
(3, 2, 4)