在python中并行化向量式计算的乘法

前端之家收集整理的这篇文章主要介绍了在python中并行化向量式计算的乘法前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我有一大堆代码

for i in range(0,len(a))
    b[i] = func(a[i])

其中a和b是相同长度的数组,a是给定的(和大),func是一些具有大量局部变量但不使用任何全局变量函数.

我想在几个cpu上分配func的计算.大概我需要使用多处理模块,但我没有找到任何相关的例子.你能帮忙吗?谢谢.

最佳答案
请参阅multiprocessing docs中的第一个代码示例:

from multiprocessing import Pool

# you could define `func`,`a` here

if __name__=="__main__":
    p = Pool() # use all available cpu cores
    b = p.map(func,a)

猜你在找的Python相关文章