Python多进程编程：为什么我的多进程代码必须放在if __name__ == "__main__":块中？

在Python中，多进程编程通常使用multiprocessing模块来实现。当你使用multiprocessing模块创建子进程时，操作系统会通过fork（在Unix-like系统上）或spawn（在Windows上）来创建新的进程。这些新进程会从头开始执行你的Python脚本。

为什么需要if __name__ == "__main__":块？

1. 避免递归创建子进程：

   • 在Windows系统上，Python使用spawn方法来创建新进程。spawn会重新导入主模块（即你的脚本），并执行其中的代码。如果没有if __name__ == "__main__":块，那么每次创建子进程时，都会重新执行整个脚本，导致无限递归地创建子进程。
   • 例如，如果你在脚本的全局作用域中直接创建子进程，那么每次子进程启动时，都会再次执行创建子进程的代码，从而导致无限递归。

2. 确保代码只在主进程中执行：

   • 通过将多进程代码放在if __name__ == "__main__":块中，你可以确保只有在主进程中执行这些代码，而不是在子进程中重复执行。这样可以避免不必要的资源消耗和潜在的错误。

示例代码

import multiprocessing

def worker_function(name):
    print(f"Worker {name} is running")

if __name__ == "__main__":
    processes = []
    for i in range(5):
        p = multiprocessing.Process(target=worker_function, args=(f"Process-{i+1}",))
        processes.append(p)
        p.start()

    for p in processes:
        p.join()

解释

• if __name__ == "__main__":：这个条件确保只有在脚本作为主程序运行时，才会执行其中的代码。如果脚本被导入为模块，则不会执行这些代码。
• multiprocessing.Process：创建子进程并执行worker_function函数。
• p.start()：启动子进程。
• p.join()：等待子进程结束。

总结

将多进程代码放在if __name__ == "__main__":块中是为了避免在Windows系统上无限递归地创建子进程，并确保代码只在主进程中执行。这是Python多进程编程中的一个重要实践，尤其是在跨平台开发时。