python编程中的if name == 'main': 的作用

最新推荐文章于 2024-07-12 10:11:14 发布

原创最新推荐文章于 2024-07-12 10:11:14 发布 · 363 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文探讨了Python编程中`if __name__ == 'main':`的作用。该语句用于区分脚本直接运行和作为模块导入的情况，确保在脚本直接执行时才执行其下的代码，防止在模块导入时意外执行主程序部分。

在搞懂感知器的基础上，很高兴地转战《零基础入门深度学习(2) - 线性单元和梯度下降》，代码见下面，却惊奇地发现：最终结果里不光有print（linear_unit）的执行结果，竟然还有print（and_perception）的执行结果，一开始以为是调用子类顺带打印了父类的__str__，后来仔细一看，不是那回事，输出的明显是上个程序的主程序部分，这样就联想到是不是自己当时图省事省去了if __name__ == 'main':造成的，在上个程序里加入这句话后，结果正确。

然后一搜看到别人分析这句代码的功能，留存吧：

一个python的文件有两种使用的方法，第一是直接作为脚本执行，第二是import到其他的python脚本中被调用（模块重用）执行。

if __name__ == 'main': 的作用就是控制这两种情况执行代码的过程：

在if __name__ == 'main': 下的代码只有在第一种情况下（即文件作为脚本直接执行）才会被执行，

而import到其他脚本中是不会被执行的。

from perceptron import Perceptron
f=lambda x:x
class LinearUnit(Perceptron):
    def __init__(self,input_num):
        Perceptron.__init__(self,input_num,f)


def train_linear_unit():
    lu = LinearUnit(1)
    input_vecs = [[5], [3], [8], [1.4], [10.1]]
    labels = [5500, 2300, 7600, 1800, 11400]
    lu.train(input_vecs, labels, 10, 0.01)
    return lu

linear_unit = train_linear_unit()
print(linear_unit)
print('Work 3.4 years, monthly salary = %.2f' % linear_unit.predict([3.4]))
print('Work 15 years, monthly salary = %.2f' % linear_unit.predict([15]))
print('Work 1.5 years, monthly salary = %.2f' % linear_unit.predict([1.5]))
print('Work 6.3 years, monthly salary = %.2f' % linear_unit.predict([6.3]))

python编程中的if __name__ == 'main': 的作用

python编程中的if name == 'main': 的作用