IPython 中的并行嵌套 for 循环

我的 python 代码中有一个嵌套的 for 循环，如下所示：

results = []

for azimuth in azimuths:
    for zenith in zeniths:
        # Do various bits of stuff
        # Eventually get a result
        results.append(result)

我想在我的 4 核机器上并行化这个循环以加快速度。查看IPython并行编程文档（http://ipython.org/ipython-doc/dev/parallel/parallel_multiengine.html#quick-and-easy-parallelism）似乎有一种简单的使用方法map并行迭代操作。

但是，要做到这一点，我需要将循环内的代码作为函数（这很容易做到），然后映射该函数。我遇到的问题是我无法获得一个数组来映射该函数。itertools.product()产生一个迭代器，我似乎无法使用它的映射函数。

我尝试在这里使用地图是不是找错了树？有更好的方法吗？或者有什么方法可以使用itertools.product然后使用映射到结果的函数进行并行执行？

要并行化每个调用，您只需要获取每个参数的列表。您可以使用itertools.product + zip得到这个：

allzeniths, allazimuths = zip(*itertools.product(zeniths, azimuths))

然后你可以使用地图：

amr = dview.map(f, allzeniths, allazimuths)

为了更深入地了解这些步骤，下面是一个示例：

zeniths = range(1,4)
azimuths = range(6,8)

product = list(itertools.product(zeniths, azimuths))
# [(1, 6), (1, 7), (2, 6), (2, 7), (3, 6), (3, 7)]

所以我们有一个“成对列表”，但我们真正想要的是每个参数都有一个列表，即“成对列表”。这正是稍微有点奇怪的地方zip(*product)语法让我们：

allzeniths, allazimuths = zip(*itertools.product(zeniths, azimuths))

print allzeniths
# (1, 1, 2, 2, 3, 3)
print allazimuths
# (6, 7, 6, 7, 6, 7)

现在我们只需将函数映射到这两个列表上，以并行化嵌套 for 循环：

def f(z,a):
    return z*a

view.map(f, allzeniths, allazimuths)

只有两个并没有什么特别之处 - 这个方法应该扩展到任意数量的嵌套循环。