首页 >后端开发 >Python教程 >`if __name__ == '__main__':` 在 Python 中做什么以及为什么它很重要?

`if __name__ == '__main__':` 在 Python 中做什么以及为什么它很重要?

Patricia Arquette
Patricia Arquette原创
2024-12-24 03:52:14747浏览

What Does `if __name__ ==

name == "__main__" 有何作用?为什么需要这样做?

当 Python 脚本作为主程序执行时,解释器将特殊变量 __name__ 赋给“__main__”。如果脚本由另一个程序作为模块导入,则 __name__ 将设置为导入模块的名称。

要理解为什么这很重要,请考虑以下示例:

# Suppose this is foo.py.

print("before import")
import math

print("before function_a")
def function_a():
    print("Function A")

print("before function_b")
def function_b():
    print("Function B {}".format(math.sqrt(100)))

print("before __name__ guard")
if __name__ == '__main__':
    function_a()
    function_b()
print("after __name__ guard")

执行流程:

  1. 特殊变量设置:

    • 名称设置为“__main__”,因为foo.py作为主程序运行。
  2. 代码执行:

    • 脚本逐行执行其所有代码:

      • 打印“导入之前”,加载数学模块等。
      • 定义函数 function_a 和function_b.
    • name 检查被评估:因为它匹配“main”,所以 if 块中的所有代码都会被执行(打印“Function A”和“Function B 10.0")。
    • 名称检查下面的代码被执行(打印“after name__guard”)。

名称的重要性检查:

此检查是至关重要,因为:

  • 保护来自意外调用: 如果 foo.py 没有 name 检查,将其作为模块导入将触发脚本使用导入脚本的命令行参数运行,这通常是不需要的。
  • Pickle 文件问题: 如果 foo.py 将自定义类保存到 pickle 文件中,请在另一个脚本会导入 foo.py 并在没有 name 检查的情况下执行其代码,从而导致上述问题。

附加说明:

  • 您可以在脚本中进行多个 name 检查,但它是不常见。
  • 使用 name 运行脚本 foo2.py 检查“a1”、“m1”、“a2”、“b”、“a3”、“m2”中包含的结果,“t2”,同时删除名称检查结果为“a1”,“a2”,“b”,“a3”, “t2”。
  • 将 foo3.py 作为脚本运行会产生“t1”、“a1”、“a2”、“b”、“a3”、“t2”,而将其作为模块结果导入
  • 在 foo4.py 中将 __name__ 设置为“__main__”(即使作为模块导入)意味着脚本在导入时将始终运行,因为 __name__ == __main__ 将始终评估为 True。

以上是`if __name__ == '__main__':` 在 Python 中做什么以及为什么它很重要?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn