/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

0 vote208 views2 answers
'dict_keys' object has no attribute 'tolist'
Python
1 vote194 views1 answer
Require http method django view
Python
0 vote177 views1 answer
What is the use of extend in python
Python
0 vote184 views1 answer
@action(detail=true) meaning
Python
11 votes194 views3 answers
If elseif in single line python
Python
1 vote230 views3 answers
Making ckeditor django responsive
Python Django
2 votes251 views1 answer
Django orm timestamp field
Python Django
0 vote209 views1 answer
Dataframe string find count
Python
0 vote194 views2 answers
How to read excel with multiple pages on pandas
Python
0 vote151 views1 answer
Pymongo dynamic structure
Python