/Python/Multiprocessing pool pass additional arguments

Multiprocessing pool pass additional arguments

#!/usr/bin/env python3
from functools import partial
from itertools import repeat
from multiprocessing import Pool, freeze_support

def func(a, b):
    return a + b

def main():
    a_args = [1,2,3]
    second_arg = 1
    with Pool() as pool:
        L = pool.starmap(func, [(1, 1), (2, 1), (3, 1)])
        M = pool.starmap(func, zip(a_args, repeat(second_arg)))
        N = pool.map(partial(func, b=second_arg), a_args)
        assert L == M == N

if __name__=="__main__":
    freeze_support()
    main()

Source: stackoverflow.com

Similar codes

9 votes96 views6 answers
How to declare variables in python
Python
0 vote90 views1 answer
Cv2 remove black borders on images
Python
3 votes106 views4 answers
Nltk in python
Python
0 vote98 views2 answers
Django run management command from code
Python
1 vote128 views2 answers
Flask bootstrap
Python Flask
-2 votes109 views1 answer
Python: raise error
Python
0 vote97 views1 answer
Asyncio\events.py", line 504, in add_reader raise NotImplementedError
Python
0 vote108 views1 answer
How to train Longistic regression with given threshold
Python
0 vote93 views2 answers
Cv2 frame size
Python
0 vote95 views1 answer
Html to json python
Python