我有一大堆代码
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)