如何使用 mpi4py 在 python 中并行化需要输出数字数组的 for 循环？

问：

我有一个for循环，它在每次迭代中计算一个数字，并将其设置为零数组中的新元素。如何使用mpi4py并行化此循环？非常感谢帮助！代码如下。

    b = 1000
    R = numpy.linspace(2, 2.5, b)
    z = numpy.zeros_like(R)
    for i in range(len(z)):
        z[i] = calculate(R[i]) #pseudocode

在这个网站上有一些并行化 for 循环的例子，它增加了数字 （[https://computationalmechanics.in/parallelizing-for-loop-in-python-with-mpi/]），但我什至不知道如何开始！（在我理解的开头部分旁边，设置等级、大小等）。