multiprocessing pool.map使用多个参数

最新推荐文章于 2025-03-13 08:10:43 发布

AI专题精讲

最新推荐文章于 2025-03-13 08:10:43 发布

阅读量1.3k

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/qq_39698985/article/details/123494802

版权

python 专栏收录该内容

62 篇文章

订阅专栏

本文通过实例解析了如何使用Python的`multiprocessing`模块实现函数的星形映射(starmap)，展示了如何利用`Pool`对象并行处理多个参数组合。主要关注`starmap`、`map`和`partial`函数在实际项目中的应用，以及它们在提高代码效率上的作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#!/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()