Python多进程pool的map方法调用多参数的函数

对多个输入值开多进程计算函数返回值

import multiprocessing
from functools import partial

#函数cal有3个参数
def cal(a,b,c):
    return a+b+c

#如果参数b和c是默认值，par相当于只有一个参数a的函数
par = partial(cal,b=1,c=2)


if __name__=='__main__':
    #进程池，核数也可自定义
    cores = multiprocessing.cpu_count()
    print(cores)
    p = multiprocessing.Pool(processes=cores)
    #输入参数列表
    l = [i for i in range(10)]
    #map调用
    res = p.map(par,l)
    print(res)

    #把输入参数和结果包成字典
    result =dict(zip(l,res))
    print(result)

运行结果：

注意的点：

1. cal函数有3个参数a,b,c， 但假定只对多个输入的参数a开多进程，b,c是默认值，可以处理成只有1个参数的函数，故利用了functools模块的partial方法
2. 参数b和c相当于被固定，传入的变化参数a必须是函数定义时的第一个参数，否则会报错
   假如，被固定的是a和c，变化的是b
   那么函数定义要写成
   def cal（b,a,c）:
   …
3. 根据网上的资料，map的原理：进程池开的进程数，代表调用计算机的内核数。可迭代对象的元素从前至后依次逐个传入函数，由各个内核分别执行，当某个内核执行完毕，继续接受下一个传入的元素