将变量传递给并行化函数

Question

我正在并行化矩阵的生成，其中矩阵中的每个元素都是由fun函数计算的。 如果我传递给该函数的唯一内容是索引i和j，则可以使它工作。 但是，我想将另一个变量传递给该函数，例如x，该怎么办？

我正在使用Python 2.7

import numpy as np                                                                  
import multiprocess as mp                                                           
import itertools                                                                    
p = mp.Pool()                                                                       

def fun((i,j)):                                                                     

   print i,j                                                                         
   prod =  i * j  
   # what if I want to have a variable x in this function
   # prod = i * j * x

   return prod                                                                       

combs = ((i,j) for i,j in itertools.product(xrange(5), repeat=2) if i <= 5)         
result = p.map(fun, combs)                                                          
p.close()                                                                           
p.join() 

newresult = np.array(result).reshape(5,5)                                           
print newresult

Answer 1

def fun((i,j,x)):                                                                     
   print i,j,x                                                                         
   prod =  i * j * x 
   return prod    

为何起作用：您实际上只是将一个对象传递给该函数，而事实证明这是一个元组。 def fun（（i，j））只是将元组与对象再次分开。 因此，要回答您的问题，您只需在元组中添加另一个元素即可，并且效果很好。

更清晰地表示您正在做什么：

def fun(data):
    i,j,x = data
    print i,j,x
    prod =  i * j * x
    return prod

data = (2,4,10)
print(fun(data))

或者您可以这样做：

def fun((i,j), x):
    print i,j, x
    prod =  i * j * x
    # what if I want to have a variable x in this function
    # prod = i * j * x

    return prod

print(fun((2,4), 10))