/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

5 votes114 views2 answers
Selenium driver wait python
Python
0 vote64 views1 answer
Pytorchvideo cant pickle local object 'kinetics data module.val dataloader.<locals>.<lambda>'
Python
6 votes108 views2 answers
Remove string punctuation python 3
Python
0 vote125 views1 answer
What is require_self
Python
2 votes127 views2 answers
How to open a website with selenium python
Python
0 vote130 views1 answer
Creating empty set and append python
Python
29 votes134 views2 answers
Python initialize array
Python
1 vote103 views1 answer
Text to sound python
Python
0 vote119 views1 answer
Python catch print
Python
0 vote122 views1 answer
Amazing Trees with python turtle
Python