有没有更好的方法向 itertools.product 发送多个参数？

问题描述

我正在尝试从包含多行的 2D 列表创建 itertools.product。例如，考虑一个列表 s:

[[0.7168573116730971,1.3404415914042531,1.8714268721791336,11.553051251803975],[0.6702207957021266,1.2476179147860895,1.7329576877705954,10.635778602978927],[0.6238089573930448,1.1553051251803976,1.5953667904468385,9.725277699842893],[0.5776525625901988,1.0635778602978927,1.4587916549764335,8.822689900641748]]

我想在列表的 4 行之间计算 itertools.product：

pr = []
for j in (it.product(s[0],s[1],s[2],s[3])):
    pr.append(j)

这给出了尺寸为 256,4 的 pr 的必要结果，其中 256 是（列数^行数）。但是，有没有更好的方法可以将 list 的所有行作为参数发送，而不必写下每一行的名称。如果要为更大的列表做这件事会很烦人。

如果 s 是一个 numpy.array，我猜可以使用 numpy.meshgrid。但即便如此，我也必须一一记下每一行作为参数。

解决方法

您可以在 Python 中使用解包符号 *：

import itertools as it

s = [[0.7168573116730971,1.3404415914042531,1.8714268721791336,11.553051251803975],[0.6702207957021266,1.2476179147860895,1.7329576877705954,10.635778602978927],[0.6238089573930448,1.1553051251803976,1.5953667904468385,9.725277699842893],[0.5776525625901988,1.0635778602978927,1.4587916549764335,8.822689900641748]]

pr = []
for j in (it.product(*s):
    pr.append(j)

它会将您列表中的每一项 s 发送到 product 函数