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中寫入如下程式碼:

1 print "I'm the first."
2 if __name__=="__main__":
3 print "I'm the second."

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

python指令碼測試


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

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

if __name__=="__main__"演示

只輸出了第一行字串。即,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__打印出來。檔案內容和結果如下,

python指令碼

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

 __name__變數

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