应用 Python 的人,平时常常会写一些脚本,不论是为了晋升工作效率,还是为了满足一些特定的需要,Python 脚本都是一个常见又有用的货色。

然而,我最近发现了一个以前未曾觉察的问题,就是脚本外面是否增加 if __name__ == "__main__": 这个语句,对脚本的应用其实是有很大影响的,并且这外面还有很大的学识。

常见误区

很多敌人在写脚本时比拟随便,简略的脚本间接一溜写下来,没有函数,程序执行。简单点的脚本,可能会加函数。这种写法可读性比拟差,常常让人一眼找不到程序运行的入口和程序。

而 Python 社区比拟举荐的写法是在写脚本时,加上上面这个语句:

def main():    # do something    print("do something.")if __name__ == "__main__":    main()

大多数人看到这里,会不会说,这有什么,加不加这个没那么重要吧!

先不要忙着不屑,让咱们一起来认真掰扯掰扯!

有什么用

在具体阐明 if __name__ == '__main__' 的作用前,先从一个简略的实例直观上感受一下。

# const.py PI = 3.14 def train():    print("PI:", PI) train()
# area.py from const import PI def calc_round_area(radius):    return PI * (radius ** 2) def calculate():    print("round area: ", calc_round_area(2)) calculate()

咱们看下 area.py 的运行后果:

PI: 3.14round area:  12.56

的 PI 变量,在运行的时候,const.py 中函数 train() 中的打印也带过去了,而咱们只是援用变量,并没有援用函数,所以这是咱们不违心看到的。

解决这个问题的办法也很简略,咱们只需在 const.py 中加上一句:

PI = 3.14def train():   print("PI:", PI)if __name__ == "__main__":   train()

再次运行 area.py ,输入后果如下:

round area:  12.56

这是咱们预期的后果。

程序运行入口

丛上述实例能够发现,如果没有 if __name__=="__main__": ,作为 area.py 导入文件时 const.py 中的所有代码都被执行了,而加上之后就只运行导入的局部代码。

这就是 if __name__=="__main__": 不言而喻的作用,实际上 if __name__=="__main__": 就相当于是 Python 模仿的程序入口。因为模块之间互相援用,不同模块可能都有这样的定义,而入口程序只能有一个,选中哪个入口程序取决于 __name__ 的值。

咱们再来看一个小程序:

# test.pyprint("look here")print(__name__) if __name__ == '__main__':    print("I'm test.py")

程序的运行后果如下:

look here__main__I'm test.py

能够发现,此时变量 __name__ 的值为 __main__,所以打印 “I'm test.py”。如果运行其余文件,通过运行的文件调用本文件,则不会打印该语句,因为程序入口不对,该语句不执行。

代码标准

有了 if __name__=="__main__": 相当于 Python 程序也有了一个入口函数,咱们能够清晰的晓得程序的逻辑开始于何处,当然还须要咱们盲目的把程序的开始逻辑都放在这里。其实,这也是 PyCharm 举荐的作法。

为什么很多优良的编程语言,比方 C、Java、Golang、C++ 都有一个 main 入口函数呢?我想很重要的一个起因就是就是程序入口对立,容易浏览。

多过程场景大作用

如果你用多过程来做并行计算,相似这样的代码:

import multiprocessing as mpdef useful_function(x):    return x * xprint("processing in parallel")with mp.Pool() as p:    results = p.map(useful_function, [1, 2, 3, 4])    print(results)

运行这段代码,控制台会始终打印:

processing in parallelprocessing in parallelprocessing in parallelprocessing in parallelprocessing in parallelprocessing in parallelprocessing in parallelprocessing in parallelprocessing in parallel

并且程序会不停的报错 RuntimeError。

如果你加上了 if __name__=="__main__": ,程序就会依照预期的进行:

import multiprocessing as mpdef useful_function(x):    return x * xif __name__ == '__main__':    print("processing in parallel")    with mp.Pool() as p:        results = p.map(useful_function, [1, 2, 3, 4])        print(results)

Python 的多程序就是启动了多个 Python 解器器,每个 Python 解释器都会导入你这个脚本,复制一份全局变量和函数给子过程用,如果有了 if __name__=="__main__":,那它前面的代码就不会被 import,也就不会被反复执行。否则,这个创立多过程的代码就会被 import,就会被执行,从而有限递归的去创立子过程

总结

if __name__=="__main__": 尽管不是强制的,然而我强列举荐你写脚本时依照这个标准来做,它是 Python 社区的约定,对应Python 之禅:明确优于费解。

以上就是本次分享的所有内容,如果你感觉文章还不错,欢送关注公众号:Python编程学习圈,每日干货分享,发送“J”还可支付大量学习材料,内容笼罩Python电子书、教程、数据库编程、Django,爬虫,云计算等等。或是返回编程学习网,理解更多编程技术常识。