Python中的 `if name == “main“` 到底有什么用？

Jia ming

489人浏览 · 2026-06-12 14:53:14

Jia ming · 2026-06-12 14:53:14 发布

一个让Python代码文件既是“工具库”又是“主程序”的神奇开关

如果你是Python初学者，在阅读别人的代码时，一定会遇到下面这个结构：

if __name__ == "__main__":
    # 一些代码...

你可能很困惑：这行代码是干什么的？为什么非要写它？不写会怎样？

今天我们就来彻底搞懂这个问题。

一句话解释

if __name__ == "__main__" 的作用是：判断当前文件是被直接运行，还是被作为模块导入到其他文件中。

如果直接运行这个文件，__name__ 会被设为 "__main__"，条件成立，代码块执行。
如果被导入到其他文件，__name__ 会是文件名，条件不成立，代码块不执行。

从例子中理解

第一步：创建一个名为 `my_math.py` 的文件

# my_math.py

def add(a, b):
    return a + b

print(f"my_math.py 的 __name__ 值是: {__name__}")

if __name__ == "__main__":
    print("这个文件是直接运行的")
    # 测试代码
    result = add(5, 3)
    print(f"测试: 5 + 3 = {result}")

第二步：直接运行它

在终端执行：

python my_math.py

输出：

my_math.py 的 __name__ 值是: __main__
这个文件是直接运行的
测试: 5 + 3 = 8

可以看到，__name__ 的值是 "__main__"，所以 if 块里的代码被执行了。

第三步：在另一个文件中导入它

创建 main.py：

# main.py
import my_math

print("这是 main.py 文件")
result = my_math.add(10, 20)
print(f"10 + 20 = {result}")

执行 main.py：

python main.py

输出：

my_math.py 的 __name__ 值是: my_math
这是 main.py 文件
10 + 20 = 30

注意：my_math.py 中的 print 依然执行了（因为它在模块顶层），但 __name__ 变成了 "my_math"，所以 if 块里的测试代码没有执行。

为什么要这样做？三个核心用途

1. 模块测试（最常用）

当你写一个模块（比如一堆数学函数）时，你可能想测试它们是否工作正常。这些测试代码只在你开发时有用，别人导入你的模块时不应该运行。

把测试代码放在 if __name__ == "__main__": 块里，完美解决这个问题。

# my_module.py
def useful_function():
    pass

if __name__ == "__main__":
    # 这些测试代码只在直接运行时执行
    # 别人导入时不会执行
    test_result = useful_function()
    print("测试通过！")

2. 定义程序的命令行入口

对于一个完整的Python程序，通常会有一个启动函数。这个结构让程序既可以被命令行直接调用，也可以被其他代码导入使用。

def main():
    """程序的主入口"""
    print("程序启动了！")
    # 主要逻辑...

if __name__ == "__main__":
    main()

这样做的好处是：

直接运行：python myprogram.py → 执行 main()
作为模块导入：import myprogram → 只导入函数和类，不会自动执行主逻辑

3. 多进程编程中的必要性

在Windows上使用Python的 multiprocessing 模块时，如果没有 if __name__ == "__main__" 保护，子进程可能会无限递归创建，导致程序崩溃。

from multiprocessing import Process

def worker():
    print("子进程工作")

if __name__ == "__main__":
    # 必须有这个保护，否则在Windows上会出问题
    p = Process(target=worker)
    p.start()