python parallel function with multi type arguments [duplicate]

Question

I have one function with multi arguments. and I need to run this function parallel for different inputs. how can I do that? I know the 'multiprocessing' module. but my problem is that for e.g pool. map accept just function with one argument. in other words, you think I have the below simple function. how can I run this function in parallel for different inputs?

def func(x,y,flag):
  if flag == 1:
     return x*y
  if flag == 2:
     return x+y
  if flag == 3:
     return x**y

Alexander Riedel · Accepted Answer · 2020-12-02 12:39:16Z

0

You can use Pool.starmap() like this

from multiprocessing import Pool

def func(x,y,flag):
  if flag == 1:
     return x*y
  if flag == 2:
     return x+y
  if flag == 3:
     return x**y
if __name__ == "__main__":
    with Pool() as p:
        res = p.starmap(func, [(1, 2, 3), (1, 2, 2),(1, 5, 1)])
        print(res)

edited Dec 2, 2020 at 12:39

answered Dec 2, 2020 at 12:33

Alexander Riedel

1,3591 gold badge10 silver badges16 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

python parallel function with multi type arguments [duplicate]

1 Answer 1

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Linked

Related