在 Python 代码中,我们经常会看到
代码语言:javascript复制if __name__ == '__main__':
本文详细总结一下用法。
先给出结论,对该语句用法简单的解释就是:如果if __name__ == '__main__' 所在模块是被直接运行的,则该语句下代码块被运行,如果所在模块是被导入到其他的python脚本中运行的,则该语句下代码块不被运行。
下边具体的讲解。
首先我们得先明白一个概念:程序入口。
程序入口
对于很多编程语言来说,程序都必须要有一个入口,比如 C,C ,以及完全面向对象的编程语言 Java,C# 等。其中C 和 C 都需要有一个 main 函数来作为程序的入口,也就是程序的运行会从 main 函数开始。同样的,Java 和 C# 则必须要有一个包含 Main 方法的主类来作为程序入口。
但和C,C 、Java 以及 C# 等有所不同的是,Python属于脚本语言,不像编译型语言那样先将程序编译成二进制再运行,python是动态的逐行解释运行,也就是从脚本第一行开始运行,没有统一的入口。
另外我们知道,python文件有两种使用方法:
- 第一种:直接作为脚本执行
- 第二种:import到其他的python脚本中被调用(模块重用)执行,也就是作为模块(库)被导入执行
不管是直接运行还是导入,最顶层的代码都会被运行(Python 用缩进来区分代码层次)。但实际使用过程中,python文件作为模块被其它python脚本调用执行的时候,有的代码我们是不希望被运行的。
而if __name__ == '__main__' 的作用就是定义这两种情况执行代码的执行方式,在该语句下的代码只有在文件作为脚本直接执行情况下才会被执行,而import到其他脚本中是不会被执行的。
案例
举个例子来说:
要计算一个圆形的面积,公式是S = л * r^2(其中r是半径)
第一步:首先创建一个 param.py 的文件,其中代码如下:
代码语言:javascript复制pi = 3.1415926
def main():
print("pi:", pi)
main()
在param.py文件里定义了一个参数 pi,直接执行该文件输出:
代码语言:javascript复制pi: 3.1415926
第二步:然后创建 calculate.py 文件,用于计算圆的面积,该文件里边需要用到 param.py 文件中的 pi 参数,需要我们从 param.py 中把 pi 变量导入到 calculate.py 中:
代码语言:javascript复制from param import pi
def area_calculate(r):
s = pi * (r ** 2)
return s
def main():
print("The area of the circle: ", area_calculate(2))
main()
运行 calculate.py(python calculate.py),输出结果:
代码语言:javascript复制pi: 3.1415926
The area of the circle: 12.5663704
从输出结果可以看出,param.py 中的 main 函数也被运行了,实际上我们不希望它被运行,这时我们就可以用if __name__ == '__main__' 对 param.py 做修改:
代码语言:javascript复制pi = 3.1415926
def main():
print("pi:", pi)
if __name__ == "__main__":
main()
再次运行param.py,输出如下:
代码语言:javascript复制pi: 3.1415926
再次运行 calculate.py,输出如下:
代码语言:javascript复制The area of the circle: 12.5663704
对比可以看出,再次运行 param.py ,if __name__ == '__main__'语句之前和之后的代码都被执行,仍然输出 pi: 3.1415926 ,但再次运行 calculate.py,导入的 param.py 里if __name__ == '__main__'语句之后的代码已经不被执行,输出已经不包含 pi: 3.1415926。
if __name__ == '__main__'就相当于是 Python 模拟的程序入口。Python 本身并没有规定这么写,这只是一种编码习惯。由于模块之间相互引用,不同模块可能都有这样的定义,而入口程序只能有一个。到底哪个入口程序被选中,这取决于__name__的值。而__name__是内置变量,用于表示当前模块的名字。
运行原理
每个python模块(python文件,也就是此处的 param.py 和 calculate.py)都包含内置的变量__name__,当运行模块被执行的时候__name__等于文件名(包含了后缀.py);如果import到其他模块中,则__name__等于模块名称(不包含后缀.py)。而'__main__'等于当前执行文件的名称(包含了后缀.py)。进而当模块被直接执行时,__name__ == '__main__'结果为真,所以后续代码可以继续执行。
举例说明。
我们在 param.py 脚本的if __name__ == '__main__'之前加入print(__name__),即将__name__打印出来。文件内容如下:
代码语言:javascript复制pi = 3.1415926
def main():
print("pi:", pi)
print(__name__)
if __name__ == "__main__":
main()
运行结果输出:
代码语言:javascript复制__main__
pi: 3.1415926
可以看出,此时变量__name__的值为'__main__'。
再执行 calculate.py,执行结果如下:
代码语言:javascript复制param
The area of the circle: 12.5663704
此时,param.py 中的__name__变量值为 param,__name__ == '__main__'条件为假,所以无法执行其后的代码。
本文转载自bigdata老司机的博客,已征得博主的同意。