生成所有在Python中尊重条件的向量

Question

是否有一种方法可以生成长度为n的所有向量的列表，其中元素介于1到6之间，如：这？我只知道如何用嵌套的for循环来处理特定的n。谢谢

Answer 1

以池和长度为单位的函数：

from itertools import product

def generate_vectors(pool, length):
    return list(product(pool, repeat=length))

generate_vectors(range(1,7), 3)
# [(1, 1, 1), (1, 1, 2), (1, 1, 3), ...]