python3编程中的if __name__ == '_
最近看一些机器学习代码,在每个函数后面都要加上
if __name__ == '__main__':
main()
一开始很不能理解,后来经过查阅一些大神的博客后终于明白了,总结下来。
一个python的文件有两种使用的方法,第一是直接作为脚本执行,第二是在其他文件的python脚本中被import即调用(模块重用)执行。因此if name == 'main':的作用就是控制这两种情况执行代码的过程,在if name == 'main':下的代码只有在第一种情况下(即文件作为脚本直接执行)才会被执行,而import到其他脚本中是不会被执行的。
来举个例子吧,在c盘下建一个a.py文件,写入如下代码
print('i first')
if __name__=='__main__':
print('i second')
在命令提示符中运行a.py结果如下
可以发现if name == 'main':前后的代码都执行了
接着我们在c盘下再新建一个b.py文件,写入
import a
运行b.py结果如下
即if name=="main":之前的语句被执行,之后的没有被执行。
tell me why?
原来每个python模块(python文件,也就是此处的a.py和b.py)都包含内置的变量name,当运行模块被执行的时候,name等于文件名(包含了后缀.py);如果模块被import到其他模块中,则name等于模块名称(不包含后缀.py)。而“main”始终等于当前执行文件的名称(包含了后缀.py)。进而当模块被直接执行时,name == 'main'结果为真。
同样举例说明,我们在a.py脚本中的if__name__=="main":之前加入print name,即将name打印出来。
print('i first')
print(__name__)
if __name__=='__main__':
print('i second')
运行结果如下
可以看出,此时变量name的值为"main"(其实name和main的值应该都是a.py);
再执行import_test.py,执行结果如下:
此时,a.py中的name变量值为a,而main的值为a.py,不满足name=="main"的条件,因此,无法执行其后的代码。