Python小技巧分享

import模組

在Python經常使用import宣告，以使用其他模組(也就是其它.py檔案)中定義的物件。

1) 使用__name__

當我們編寫Python庫模組的時候，我們往往執行一些測試語句。當這個程式作為庫被import的時候，我們並不需要執行這些測試語句。一種解決方法是在import之前，將模組中的測試語句註釋掉。Python有一種更優美的解決方法，就是使用__name__。

下面是一個簡單的庫程式TestLib.py。當直接執行TestLib.py時，__name__為"__main__"。如果被import的話，__name__為"TestLib"。

def lib_func(a):
    return a + 10
def lib_func_another(b):
    return b + 20
if __name__ == '__main__':
    test = 101
    print(lib_func(test))

我們在user.py中import上面的TestLib。

import TestLib
print(TestLib.lib_func(120))

你可以嘗試不在TestLib.py中使用if __name__=='__main__'， 並對比執行結果。

 

2) 更多import使用方式

import TestLib as test         # 引用TestLib模組，並將它改名為t

比如:

import TestLib as t
print(t.lib_func(120))

from TestLib import lib_func   # 只引用TestLib中的lib_func物件，並跳過TestLib引用欄位

這樣的好處是減小所引用模組的記憶體佔用。

比如：

from TestLib import lib_func
print(lib_func(120))

from TestLib import *          # 引用所有TestLib中的物件，並跳過TestLib引用欄位

比如:

from TestLib import *
print(lib_func(120))

查詢

1) 查詢函式的引數

當我們想要知道某個函式會接收哪些引數的時候，可以使用下面方法查詢。

import inspect
print(inspect.getargspec(func))

2) 查詢物件的屬性

除了使用dir()來查詢物件的屬性之外，我們可以使用下面內建(built-in)函式來確認一個物件是否具有某個屬性：

hasattr(obj, attr_name)   # attr_name是一個字串

例如：

a = [1,2,3]
print(hasattr(a,'append'))

2) 查詢物件所屬的類和類名稱

a = [1, 2, 3]
print a.__class__
print a.__class__.__name__

3) 查詢父類

我們可以用__base__屬性來查詢某個類的父類：

cls.__base__

例如：

print(list.__base__)

使用中文(以及其它非ASCII編碼)

在Python程式的第一行加入#coding=utf8，例如:

#coding=utf8
print("你好嗎？")

也能用以下方式：

#-*- coding: UTF-8 -*-
print("你好嗎？")

表示2進位制，8進位制和16進位制數字

在2.6以上版本，以如下方式表示

print(0b1110)     # 二進位制，以0b開頭
print(0o10)       # 八進位制，以0o開頭
print(0x2A)       # 十六進位制，以0x開頭

如果是更早版本，可以用如下方式：

print(int("1110", 2))
print(int("10", 8))
print(int("2A", 16))

註釋

一行內的註釋可以以#開始

多行的註釋可以以'''開始，以'''結束，比如

'''
This is demo
'''
def func():
    # print something
    print("Hello world!")  # use print() function
# main
func()

註釋應該和所在的程式塊對齊。

搜尋路徑

當我們import的時候，Python會在搜尋路徑中查詢模組(module)。比如上面import TestLib，就要求TestLib.py在搜尋路徑中。

我們可以透過下面方法來檢視搜尋路徑：

import sys
print(sys.path)

我們可以在Python執行的時候增加或者刪除sys.path中的元素。另一方面，我們可以透過在shell中增加PYTHONPATH環境變數，來為Python增加搜尋路徑。

下面我們增加/home/vamei/mylib到搜尋路徑中：

$export PYTHONPATH=$PYTHONPATH:/home/vamei/mylib

你可以將正面的這行命令加入到～/.bashrc中。這樣，我們就長期的改變了搜尋路徑。

 

指令碼與命令列結合

可以使用下面方法執行一個Python指令碼，在指令碼執行結束後，直接進入Python命令列。這樣做的好處是指令碼的物件不會被清空，可以透過命令列直接呼叫。

$python -i script.py

 

安裝非標準包

Python的標準庫隨著Python一起安裝。當我們需要非標準包時，就要先安裝。

1) 使用Linux repository (Linux環境)

這是安裝Python附加包的一個好的起點。你可以在Linux repository中查詢可能存在的Python包 (比如在Ubuntu Software Center中搜尋matplot)。

2) 使用pip。pip是Python自帶的包管理程式，它連線Python repository，並查詢其中可能存在的包。

比如使用如下方法來安裝、解除安裝或者升級web.py：

$pip install web.py

$pip uninstall web.py

$pip install --upgrade web.py

如果你的Python安裝在一個非標準的路徑(使用$which python來確認python可執行檔案的路徑)中，比如/home/vamei/util/python/bin中，你可以使用下面方法設定pip的安裝包的路徑:

$pip install --install-option="--prefix=/home/vamei/util/" web.py

3) 從原始碼編譯

如果上面方法都沒法找到你想要的庫，你可能需要從原始碼開始編譯。Google往往是最好的起點。