代码之家 › 专栏 › 技术社区 › SuperCiocia

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

python

SuperCiocia · 技术社区 · 6 年前

我有以下代码:

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 所以我必须给出一个关于重复次数的论点。。。

我试着给 产品 一个列表和一个元组,这是行不通的。

有什么想法吗?

3 回复 | 直到 6 年前

chepner 6 年前

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

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

Daniel 6 年前

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)

Louis Ng 6 年前

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

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

6 月前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

6 月前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

6 月前

user29715306 · from_users=和chats=电视节目中的差异

6 月前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

6 月前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

6 月前

prayner · 更新嵌套字典包含列表中的项

6 月前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

6 月前

Dave · 如何在for循环中修改列表值

6 月前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

6 月前