1、这段代码的功能
一个 Python 的文件有两种使用的方法,第一是直接作为脚本执行,第二是 import 到其他的 Python 脚本中被调用(模块重用)执行。因此 if __name__ == '__main__':
的作用就是控制这两种情况执行代码的过程,在 if __name__ == '__main__':
下的代码只有在第一种情况下(即文件作为脚本直接执行)才会被执行,而 import 到其他脚本中是不会被执行的。
举个例子,下面在 test.py 中写入如下代码:
print "I'm the first."
if __name__=='__main__':print "I'm the second."
直接执行 test.py 后结果如下,可以成功打印两行字符串。即 if __name__=='__main__':
语句之前和之后的代码都被执行:
I'm the first.
I'm the second.
然后在同一文件夹新建名称为 import_test.py 的脚本,只输入以下代码:
import test
执行 import_test.py 脚本,只输出了第一行字符串。即if __name__=='__main__':
之前的语句被执行,之后的没有被执行:
I'm the first.
2、运行的原理
每个Python模块(Python文件,也就是此处的 test.py 和 import_test.py)都包含内置的变量 __name__
,当运行模块被执行的时候,__name__
等于文件名(包含了后缀.py
);如果 import 到其他模块中,则 __name__
等于模块名称(不包含后缀.py
)。而 '__main__'
等于当前执行文件的名称(包含了后缀.py
)。进而当模块被直接执行时,__name__ == 'main'
结果为真。
同样举例说明,我们在 test.py 脚本的if __name__=='__main__':
之前加入print __name__
,即将__name__
打印出来:
print "I'm the first."
print __name__
if __name__=='__main__':print "I'm the second."
运行结果如下:
I'm the first.
__main__
I'm the second.
可以看出,此时变量__name__
的值为 __main__
;
再执行 import_test.py,模块内容和执行结果如下:
import test
I'm the first.
test
此时,test.py 中的 __name__
变量值为 test
,不满足 __name__=="__main__"
的条件,因此,无法执行其后的代码。