Python小技巧

在這里列舉一些我使用Python時積累的小技巧。這些技巧是我在使用Python過程中經常使用的。之前很零碎的記在筆記本中，現在整理出來，和大家分享，也作為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往往是最好的起點。