Python-有没有一种方法可以自动执行<代码>itertools.product</code>

product 接受一个可选的整数参数,指定要重复iterable参数的次数。

np.array(product(x, repeat=2))
np.array(product(x, repeat=3))
np.array(product(x, repeat=4))
# etc

product 对努比来说,有一个论点 repeat x 必须重复:

def np_product(x, repeat):
    result = np.ndarray((len(x),)*repeat + (repeat,))
    for n in range(repeat):
        index = (None,) * n + (slice(None),) + (None,) * (repeat-n-1)
        result[..., n] = x[index]
    return result.reshape(-1, repeat)

a = np_product(x, repeat)

n = 4
lst = [x for _ in range(n)]
[i for i in product(*lst)]