Python 小技巧：冻结参数，让你的代码变简洁

2023-12-06
福建
本文字数：1064 字
阅读完需：约 3 分钟

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