распараллеливания умножение векторов-подобные вычислений в питоне

Question

У меня есть кусок кода, как

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

, где а и Ь массивы одинаковой длины, а даются (и большой), функ есть некоторая функция который имеет множество локальных переменных, но не использует глобальных переменных.

Я хотел бы распространять вычисления func на нескольких процессорах. Предположительно мне нужно использовать модуль многопроцессорности, но я не нашел подходящих примеров. Не могли бы вы помочь? Благодарю.

Answer 1

См самый первый пример кода в 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) 

Answer 2

Использование процесса пула. Вы можете увидеть полный образец в моем github: https://github.com/mariazverina/codejam/blob/master/src/main.py

from multiprocessing import Pool 

p = Pool(4) # set to number of cores 
b = p.map(func, a)