/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

1 vote194 views1 answer
Not null constraint failed django
Python Django
5 votes175 views1 answer
Selenium webdriver imports
Python
3 votes182 views1 answer
Get column index pandas
Python
6 votes182 views2 answers
Django read mesage
Python
30 votes171 views4 answers
UnicodeDecodeError: 'utf-8' codec can't decode byte invalid start byte
Python
0 vote164 views1 answer
Not mutable data type in python
Python
1 vote174 views2 answers
Python return using if
Python
0 vote227 views1 answer
Add a new categorical column to an existing table python
Python
0 vote167 views2 answers
Django serializer method field read write
Python
-1 vote171 views2 answers
Python choose function
Python