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

Question

有没有办法生成长度为n的所有向量的列表，元素在1到6之间，如下所示：this？我只知道如何使用嵌套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), ...]