Question

例如，如果q = 2，那么我必须生成[1,1]到[2,2]之间的所有序列。如果q = 3，则生成[1,1,1]到[3,3,3]之间的序列。对于q = 4，然后生成[1,1,1,1]到[4,4,4,4]之间的序列等。

序列的例子。对于q = 3

(1, 1, 1)
(1, 1, 2)
(1, 1, 3)
(1, 2, 1)
(1, 2, 2)
(1, 2, 3)
(1, 3, 1)
(1, 3, 2)
(1, 3, 3)
(2, 1, 1)
(2, 1, 2)
(2, 1, 3)
(2, 2, 1)
(2, 2, 2)
(2, 2, 3)
(2, 3, 1)
(2, 3, 2)
(2, 3, 3)
(3, 1, 1)
(3, 1, 2)
(3, 1, 3)
(3, 2, 1)
(3, 2, 2)
(3, 2, 3)
(3, 3, 1)
(3, 3, 2)
(3, 3, 3)

我试过这个“Python generating all nondecreasing sequences”但没有得到所需的输出。

目前我正在使用此代码，

import itertools

def generate(q):
    k = range(1, q+1) * q
    ola = set(i for i in itertools.permutations(k, q))
    for i in sorted(ola):
        print i

generate(3)

我需要另一种好方法来生成这个序列。

Answer 1

将itertools.product与repeat参数一起使用：

q = 2
list(itertools.product(range(1, q + 1), repeat=q))
Out: [(1, 1), (1, 2), (2, 1), (2, 2)]

q = 3

list(itertools.product(range(1, q + 1), repeat=q))
Out: 
[(1, 1, 1),
 (1, 1, 2),
 (1, 1, 3),
 (1, 2, 1),
 (1, 2, 2),
 ...

Answer 2

我认为你想要itertools.product()，它可以完成可迭代元素的所有可能组合。 itertools.permutations()不重复元素，itertools.combinations()或itertools.combinations_with_replacement()仅按排序顺序排列（例如，输入迭代的第一个元素不会成为结果的最后一个元素）。

from itertools import product

def generate(q):
    assert q > 0  # not defined for <= 0
    return list(product(range(1,q+1), repeat=q))

generate(3)  # [(1,1,1), (1,1,2), ..., (3,3,2), (3,3,3)]

请参阅：https://docs.python.org/3/library/itertools.html

如何使用python生成此序列

2 个答案: