python - 我在循环中调用一个函数并将结果存储在一个矩阵中，我该如何并行化它？

Question

我是 python 新手，我有一个代码，我在其中调用了一个特殊函数 g 次，并将该函数的结果恢复为最终矩阵的两列，所以在这个“for 循环”结束时，我有一个矩阵10 列，存储 5 次调用的结果，

你能指导我如何用 5 个核心来做这 5 个调用吗？

我要并行化的部分代码是：

for r in range(5):
    Final[:,2*r:2*r+2]=my_function(X)

我做了：

Final[:,2*rank:2*rank+2]=my_function(X)

但是做了之后mpiexec -np 4 python comm.py我才明白这还不够，必须有一些其他的东西，比如Barrier(）或者send() recive()我不知道，......请你指导我吗？

标签： pythonmpi4py