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")
执行流程:
-
特殊变量设置:
-
名称设置为“__main__”,因为foo.py作为主程序运行。
-
代码执行:
-
脚本逐行执行其所有代码:
- 打印“导入之前”,加载数学模块等。
- 定义函数 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中文网其他相关文章!