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

开发 前端
今天我们来说说 partial 函数, partial 函数看起来平平无奇,如果你经常翻看高手们写的库,会发现很多地方都在使用这函数。

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 生成的对象供外部的调用者直接使用。

责任编辑:赵宁宁 来源: 数据大宇宙
相关推荐

2024-11-11 17:00:27

字典压缩Python代码

2022-05-10 10:28:21

JavaScript代码

2024-01-08 17:09:07

Python解释器CPython

2022-01-06 22:31:21

Python技巧代码

2020-05-07 17:03:49

Python编码开发

2023-03-09 15:45:36

ES6编码技巧数组

2023-11-23 13:50:00

Python代码

2020-07-08 17:06:00

Python开发工具

2024-09-18 05:00:00

Python代码

2019-04-29 08:31:25

PythonPandas数据

2019-07-31 10:24:16

JavaScript浏览器口袋妖怪

2020-08-06 16:34:48

Python开发工具

2024-10-08 10:24:41

Python编程语言

2024-12-19 09:05:13

Python链式调用

2023-02-23 16:49:11

ES6技巧

2009-10-27 09:09:06

Eclipse技巧

2020-08-06 00:25:38

Python代码开发

2020-05-07 09:02:55

代码Python开发

2024-05-30 14:21:00

lambdaPython代码

2020-05-06 16:32:18

for循环Python迭代
点赞
收藏

51CTO技术栈公众号