Python-有没有办法自动执行此操作? <code>itertools.product</code>

时间:2018-10-13 16:56:22

标签: python

我有以下代码:

import numpy as np
from itertools import product
x = np.arange(-1, 2)
a = np.array([i for i in product(x,x,x,x)])

这也是我还需要np.array([i for i in product(x,x)])np.array([i for i in product(x,x,x)]) ...所以我想自动化product,这样我只需要给出重复次数的参数即可...

我试图给product列出一个列表和一个元组,这不起作用。

有什么想法吗?

3 个答案:

答案 0 :(得分:2)

为自己的numpy编写const log = require('../lib/Logger').getChildLogger('foo') log.info('Foo') ,并使用参数const log = require('./lib/Logger').getChildLogger('foo') log.info('Bar') ,即必须重复执行product的频率:

repeat

答案 1 :(得分:1)

product带有一个可选的整数参数,用于指定要重复迭代参数的次数。

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

答案 2 :(得分:0)

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

您可以*后跟* args列表