Python 中 if __name__ == '__main__': 的理解

目录

1、这段代码的功能

2、运行的原理

---

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__" 的条件，因此，无法执行其后的代码。