Python/ target=_blank class=infotextkey>Python 有一些非常使用的模塊, functools 就是其中之一。今天我們來說說其中的 partial 函數, partial 函數看起來平平無奇,如果你經常翻看高手們寫的庫,會發現很多地方都在使用這函數。
入門
我們從一個小場景開始。
現在我們需要讀取一個 txt 文件:
with open('文件1.txt',mode='r',encoding='utf8') as f:
print(f.readline())
如果同時加載2個文件:
with open('文件1.txt',mode='r',encoding='utf8') as f:
print(f.readline())
with open('文件2.txt',mode='r',encoding='utf8') as f:
print(f.readline())
初學者都能看出來,這代碼很多重復的內容,怎么簡化?
方式1:
def my_open_file(file):
return open(file,mode='r',encoding='utf8')
with my_open_file('文件1.txt') as f:
print(f.readline())
with my_open_file('文件2.txt') as f:
print(f.readline())
思路很簡單,把重復的函數整個提到自定義函數里面,把變化的文件路徑定義成參數。
方式2:
open_kws = {'mode':'r','encoding':'utf8'}
with open('文件1.txt',**open_kws) as f:
print(f.readline())
with open('文件2.txt',**open_kws) as f:
print(f.readline())
利用 python 的字典解包入參,從參數上提取。
這些方式都略顯麻煩。
方式2有其應用場景,以后再展開講解
凍結參數
"出場儀式"還是需要的:
from functools import partial
類似前面的方式1, partial 函數提供了一個更簡潔的方式
my_open_file = partial(open,mode='r',encoding='utf8')
with my_open_file('文件1.txt') as f:
print(f.readline())
with my_open_file('文件2.txt') as f:
print(f.readline())
- partial 第一個參數傳入需要"凍結參數" 的目標函數 open
- 后面接的參數,就像在調用 open 函數一樣設置即可
上面我們使用關鍵字設置參數,當然也可以按位置傳入參數,或者混合使用。
def show_nums(a,b,c):
print(a,b,c)
show_free_c = partial(show_nums,1,2)
show_free_c(3) # >>> 1 2 3
show_free_b = partial(show_nums,1,c=3)
show_free_b(2) # >>> 1 2 3
接下來就要說說 partial 的小缺點。
不是函數
partial 生成的是一個 partial 對象,他不是函數對象,這意味著它生成出來的東西丟失了函數說明。
現在為之前例子中的函數加上一些注釋:
可以看到在 vscode 上可以正常顯示信息。但是經過 partial 得到的新函數卻什么都看不到:
所以一般情況下,我們都是在一個局部小范圍內使用 partial ,比如在自定義模塊內部使用,盡可能避免讓 partial 生成的對象供外部的調用者直接使用。