将参数发送到另一个脚本的main函数

Question

我想将函数和参数发送到其他脚本的main函数中。例如在a.py我有：

#a.py

if __name__=='__main__':
    p=multiprocessing.Pool(4)
    results=p.map(task,args)

在脚本b.py中我有：

def task(x):
    return x*x
args=range(10)

如何在不修改task 的情况下，将args b.py a.py发送到a.py 的主要功能？也就是说，我想将函数和参数发送到模块中，以便在不修改此模块的情况下进行并行编程。

Answer 1

在a.py中，在插入任何代码之前，添加以下行：

import b

或者，如果您只想导入功能task和args，请使用：

from b import task
from b import args

确保a.py和b.py位于同一目录中。