/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 vote355 views1 answer
Python scatter plot legend
Python
2 votes351 views2 answers
Save a Dictionary to File in Python Using the dump Function of the pickle Module
Python
1 vote351 views1 answer
Python tomorrow
Python
9 votes389 views1 answer
Install matplotlib conda
Python
62 votes336 views5 answers
How to pick a random number in a list python
Python
0 vote337 views1 answer
Switch between frames in tkinter
Python
7 votes324 views1 answer
Discord.py ban
Python
0 vote291 views1 answer
Keyboard input raspberry pi
Python
5 votes367 views3 answers
Multiple variables in for loop python
Python
1 vote358 views6 answers
Code grepper
Python