快捷導(dǎo)航

Python?functools凍結(jié)參數(shù)小技巧實(shí)現(xiàn)代碼簡(jiǎn)潔優(yōu)化

更新時(shí)間：2023年12月11日 10:03:31 作者：高端章魚哥

這篇文章主要為大家介紹了Python?functools凍結(jié)參數(shù)小技巧實(shí)現(xiàn)代碼簡(jiǎn)潔優(yōu)化示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪

Python functools

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

使用場(chǎng)景

我們從一個(gè)小場(chǎng)景開始。

現(xiàn)在我們需要讀取一個(gè) txt 文件：

with open('文件1.txt',mode='r',encoding='utf8') as f:
    print(f.readline())

如果同時(shí)加載 2 個(gè)文件：

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())

初學(xué)者都能看出來(lái)，這代碼很多重復(fù)的內(nèi)容，怎么簡(jiǎn)化？

簡(jiǎn)化方式

方式 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())

思路很簡(jiǎn)單，把重復(fù)的函數(shù)整個(gè)提到自定義函數(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 的字典解包入?yún)?，從參?shù)上提取。

這些方式都略顯麻煩。

方式 2 有其應(yīng)用場(chǎng)景，以后再展開講解。

凍結(jié)參數(shù)

"出場(chǎng)儀式"還是需要的：

from functools import partial

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

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

上面我們使用關(guān)鍵字設(shè)置參數(shù)，當(dāng)然也可以按位置傳入?yún)?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

接下來(lái)就要說(shuō)說(shuō) partial 的小缺點(diǎn)。

不是函數(shù)

partial 生成的是一個(gè) partial 對(duì)象，他不是函數(shù)對(duì)象，這意味著它生成出來(lái)的東西丟失了函數(shù)說(shuō)明。

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

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

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

以上就是Python functools凍結(jié)參數(shù)小技巧實(shí)現(xiàn)代碼簡(jiǎn)潔優(yōu)化的詳細(xì)內(nèi)容，更多關(guān)于Python functools優(yōu)化代碼的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: