/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 vote146 views1 answer
How to write stuff in python
Python
1 vote146 views2 answers
Java Array objects
Python
1 vote145 views1 answer
Panda dataframe read csv change string to float
Python
1 vote150 views1 answer
Parse youtube video id from youtube link python
Python
0 vote153 views1 answer
ImportError: cannot import name 'StringIO'
Python
8 votes161 views2 answers
How to earse special chrat¥cter from string in python
Python
3 votes150 views3 answers
Count missing values by column in pandas
Python
2 votes159 views1 answer
Pandas drop extension name from list of files
Python
14 votes151 views2 answers
You-tube
Python
10 votes134 views3 answers
Count occurrence in array python
Python