python之__name__的妙用

引用：python中 “name” 详解

1、一个示例

# draft.py
def getInfo(a, b):
    print("__name__:",__name__)
    print(a + b)


if __name__ == '__main__':
    getInfo(2, 3)

# test.py
from draft import getInfo

getInfo(6,7)

运行draft.py，输出：

__name__: __main__
5

运行test.py，输出：

__name__: draft
13

至此可以看出，当直接运行draft时，脚本中的__name__为__main__，若是被import时，其中的__name__为draft

2、妙用

当我们定义一个模型结构时，往往会在这个类下实例化一个模型经行测试，如果将实例化测试的代码写在if __name__=="__main__":下，则直接运行该代码会运行测试部分，当我们在其他地方import时，则不会运行该测试部分。