1. 程式人生 > >python程式設計中的if __name__ == 'main' 的作用和原理

python程式設計中的if __name__ == 'main' 的作用和原理

在大多數編排得好一點的指令碼或者程式裡面都有這段if __name__ == 'main': ,雖然一直知道他的作用,但是一直比較模糊,收集資料詳細理解之後與大家分享一下。

1、程式碼的功能

   一個python的檔案有兩種使用的方法,第一是直接作為指令碼執行,第二是import到其他的python指令碼中被呼叫(模組重用)執行。因此if __name__ == 'main': 的作用就是控制這兩種情況執行程式碼的過程,在if __name__ == 'main': 下的程式碼只有在第一種情況下(即檔案作為指令碼直接執行)才會被執行,而import到其他指令碼中是不會被執行的。

舉個例子,下面在test.py中寫入如下程式碼:

print "I'm the first."
print "I'm the first."
if __name__=="__main__":
    print "I'm the second."
並直接執行test.py,結果如下圖,可以成功print兩行字串。即,if __name__=="__main__": 語句之前和之後的程式碼都被執行。

然後在同一資料夾新建名稱為import_test.py的指令碼,只輸入如程式碼:

import test

執行import_test.py指令碼,輸出結果如下:

由上圖可以看出指令碼只輸出了第一行字串。即,if __name__=="__main__": 之前的語句被執行,之後的沒有被執行。

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."

可以看出,此時變數__name__的值為"__main__";   再執行import_test.py,模組內容和執行結果如下:

此時,test.py中的__name__變數值為test,不滿足__name__=="__main__"的條件,因此,無法執行其後的程式碼。