寫Python時的5個壞習慣,你有幾條?
很多文章都有介紹怎么寫好 Python,我今天呢,相反,說說寫代碼時的幾個壞習慣。有的習慣會讓 Bug 變得隱蔽難以追蹤,當然,也有的并沒有錯誤,只是個人覺得不夠優雅。
注意:示例代碼在 Python 3.6 環境下編寫
1 用列表作函數的默認參數
看下面這個例子
- def func(a, b=[]):
- b.append(a)
- print(f'a: {a}')
- print(f'b: {b}')
- func(1)
- func(2)
正常我們期望的結果應該是這樣的
- a: 1
- b: [1]
- a: 2
- b: [2]
但當我們執行代碼后,只會得到這樣的結果
- a: 1
- b: [1]
- a: 2
- b: [1, 2]
與預期不一致。為什么呢?因為 Python 列表是可變對象,而且函數傳參又是傳的引用,所以當第二次調用 func 方法前,b 中已經有了元素 1,調用后 b 最終有兩個元素 1 和 2。
示例中 func 方法比較簡單,當發現問題的時候簡單看下就能找到根源。但是,如果是在一個比較復雜的方法里面,你有可能會粗心的忽略這一點,從而會碰到一些莫名其妙的問題。
所以,當我們要為函數設置默認參數的時候,不要使用可變對象。
上面的代碼改成這樣就 OK 了
- def func(a, b=None):
- if b is None:
- b = []
- b.append(a)
- print(f'a: {a}')
- print(f'b: {b}')
執行后得到預期結果
- a: 1
- b: [1]
- a: 2
- b: [2]
2 文件操作
很多剛接觸 Python 的伙伴做文件操作的時候很容易寫類似的代碼
- file = open('file_name')
- try:
- for line in file:
- print(line)
- finally:
- file.close()
這沒有問題,不過文件資源我們沒有必要手動去維護,像關閉這樣的操作交給上下文管理器做就好。
- with open('file_name') as file:
- for line in file:
- print(line)
這樣看起來不是清爽很多。
3 捕獲所有異常
- try:
- pass # 做一些操作
- except Exception as e:
- print(f'Exception {e}')
就像上面一樣,有時我們為了能夠快速的完成功能,很容易不管三七二十一,就捕獲 Exception 異常。這可能會捕捉到鍵盤中斷(KeyboardInterrupt)(CTRL + C)或斷言錯誤(AsstionError)等異常。捕獲不確定的異常,有時也會讓我們的程序出現莫名其妙的問題,我們應該避免這樣做。
準確的做法是根據上下文捕獲 ValueError 、AttributeError 、TypeError 等比較具體的異常,然后做適當的錯誤處理,比如打印日志等。
4 忽略 Python 的 for...else 語法
開發中我們很容易碰到類似的需求,在一個列表中,確定某個特定的元素是否存在。比如,下面的代碼便是確定列表中有沒有奇數存在
- numbers = [1, 2, 3, 4, 5]
- is_odd_exist = False
- for n in numbers:
- if n % 2 == 1:
- is_odd_exist = True
- break
- if is_odd_exist:
- print('Odd exist')
- else:
- print('Odd not exist')
這里,我們使用了一個標識 is_odd_exist,默認為 False。當找到奇數時,將其置為 True,然后跳出循環。這樣寫并沒有問題,但是我們可以換種方式
- numbers = [1, 2, 3, 4, 5]
- for n in numbers:
- if n % 2 == 1:
- print('Odd exist')
- break
- else:
- print('Odd not exist')
先介紹下 Python 的 for...else 語法,當 for 循環是正常結束時(即不是通過 break 跳出結束的),會執行 else 中的語句。
這里,我們使用了相對于其他語言如 C、PHP 等不同的一種方式,完成了相同的功能,看起來代碼也簡潔了不少。
5 使用鍵遍歷字典
初學 Python 的伙伴,可能容易寫出這樣的代碼
- member = {'name': 'xiaoming',
- 'age': 18,
- 'mobile': '18312341234'}
- for key in member:
- print(f'{key}: {member[key]}')
同樣,這也是沒有問題的,但看起來并不直觀。字典遍歷的時候,其實可以直接取出鍵值信息,像這樣
- member = {'name': 'xiaoming',
- 'age': 18,
- 'mobile': '18312341234'}
- for key, val in member.items():
- print(f'{key}: {val}')
這樣的話,看起來要明了一些。
上面提到的幾點有些帶有自己一定的偏見,不要求大家都接受,選擇合理的使用就好。