/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

8 votes337 views6 answers
Python string formatting
Python
1 vote299 views1 answer
Python pattern matching
Python
42 votes318 views4 answers
Import datetime
Python
101 votes334 views6 answers
Remove last character from string python
Python
0 vote325 views2 answers
How to delete a row based on a criteria in python datafram
Python
0 vote312 views1 answer
From_bytes python
Python
1 vote327 views1 answer
Debconf: falling back to frontend: Readline Configuring tzdata
Python
13 votes274 views2 answers
Year to datetime python
Python
0 vote263 views3 answers
How to compare the two key from constant value to list of string in python
Python
3 votes300 views1 answer
Print the heat map python
Python