Python 有一些非常使用的模塊， functools 就是其中之一。今天我們來說說其中的 partial 函數(shù)， partial 函數(shù)看起來平平無奇，如果你經常翻看高手們寫的庫，會發(fā)現(xiàn)很多地方都在使用這函數(shù)。

入門

我們從一個小場景開始。

現(xiàn)在我們需要讀取一個 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())

思路很簡單，把重復的函數(shù)整個提到自定義函數(shù)里面，把變化的文件路徑定義成參數(shù)。

方式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 的字典解包入參，從參數(shù)上提取。

這些方式都略顯麻煩。

方式2有其應用場景，以后再展開講解

凍結參數(shù)

"出場儀式"還是需要的：

from functools import partial

類似前面的方式1， partial  函數(shù)提供了一個更簡潔的方式

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 第一個參數(shù)傳入需要"凍結參數(shù)" 的目標函數(shù) open
• 后面接的參數(shù)，就像在調用 open 函數(shù)一樣設置即可

上面我們使用關鍵字設置參數(shù)，當然也可以按位置傳入參數(shù)，或者混合使用。

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 的小缺點。

不是函數(shù)

partial 生成的是一個 partial 對象，他不是函數(shù)對象，這意味著它生成出來的東西丟失了函數(shù)說明。

現(xiàn)在為之前例子中的函數(shù)加上一些注釋：

可以看到在 vscode 上可以正常顯示信息。但是經過 partial 得到的新函數(shù)卻什么都看不到：

所以一般情況下，我們都是在一個局部小范圍內使用 partial ，比如在自定義模塊內部使用，盡可能避免讓 partial 生成的對象供外部的調用者直接使用。

分享標題：Python小技巧：凍結參數(shù)，讓你的代碼變簡潔
文章分享：http://uogjgqi.cn/article/dpidjgg.html