Python 函数怎样重载
2023-04-18 14:54:41 时间
什么是函数重载?简单的理解,支持多个同名函数的定义,只是参数的个数或者类型不同,在调用的时候,解释器会根据参数的个数或者类型,调用相应的函数。
重载这个特性在很多语言中都有实现,比如 C++、Java 等,而 Python 并不支持。这篇文章呢,通过一些小技巧,可以让 Python 支持类似的功能。
参数个数不同的情形
先看看这种情况下 C++ 是怎么实现重载的
#include <iostream>
using namespace std;
int func(int a)
{
cout << 'One parameter' << endl;
}
int func(int a, int b)
{
cout << 'Two parameters' << endl;
}
int func(int a, int b, int c)
{
cout << 'Three parameters' << endl;
}
如果 Python 按类似的方式定义函数的话,不会报错,只是后面的函数定义会覆盖前面的,达不到重载的效果。
>>> def func(a):
... print('One parameter')
...
>>> def func(a, b):
... print('Two parameters')
...
>>> def func(a, b, c):
... print('Three parameters')
...
>>> func(1)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: func() missing 2 required positional arguments: 'b' and 'c'
>>> func(1, 2)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: func() missing 1 required positional argument: 'c'
>>> func(1, 2, 3)
Three parameters
但是我们知道,Python 函数的形参十分灵活,我们可以只定义一个函数来实现相同的功能,就像这样
>>> def func(*args):
... if len(args) == 1:
... print('One parameter')
... elif len(args) == 2:
... print('Two parameters')
... elif len(args) == 3:
... print('Three parameters')
... else:
... print('Error')
...
>>> func(1)
One parameter
>>> func(1, 2)
Two parameters
>>> func(1, 2, 3)
Three parameters
>>> func(1, 2, 3, 4)
Error
参数类型不同的情形
同样,先看下当前情况下 C++ 的重载是怎么实现的
#include <iostream>
using namespace std;
int func(int a)
{
cout << 'Int: ' << a << endl;
}
int func(float a)
{
cout << 'Float: ' << a << endl;
}
代码中,func 支持两种类型的参数:整形和浮点型。调用时,解释器会根据参数类型去寻找合适的函数。Python 要实现类似的功能,需要借助 functools.singledispatch 装饰器。
from functools import singledispatch
@singledispatch
def func(a):
print(f'Other: {a}')
@func.register(int)
def _(a):
print(f'Int: {a}')
@func.register(float)
def _(a):
print(f'Float: {a}')
if __name__ == '__main__':
func('zzz')
func(1)
func(1.2)
func 函数被 functools.singledispatch 装饰后,又根据不同的参数类型绑定了另外两个函数。当参数类型为整形或者浮点型时,调用绑定的对应的某个函数,否则,调用自身。
执行结果
Other: zzz
Int: 1
Float: 1.2
需要注意的是,这种方式只能够根据第一个参数的类型去确定最后调用的函数。
关于 singledispatch 的更多细节请看官方文档
https://docs.python.org/3.6/library/functools.html#functools.singledispatch
注意:函数返回值不同也是重载的一种情况,暂时没有比较好的 Python 实现方式,所以没有提及
个人觉得,重载就是为了语言的灵活性而设计的,而 Python 函数本来就有不少巧妙的设计,这个时候去仿这个技术,其实没有多大必要,而且感觉有些违背 Python 的哲学。所以,本文更多的是在讲如何模仿,而对于重载的使用场景并没有作多少说明。
觉得有帮助的麻烦点个赞~谢谢阅读。
相关文章
- Python使用tkinter组件Label显示简单数学公式
- 内网渗透之DCOM横向移动
- 以目标为导向的语义交流的共同语言——一个课程学习框架
- python爬虫前奏【成信笔记】
- HTML 5 File API:文件拖放上传功能
- 教你快速创建 Python 虚拟环境
- pyenv 实现Python多版本自由切换
- 用 Python 对 Excel文件进行批量操作
- Python - 接入钉钉机器人
- Python - 抓取 iphone13 pro 线下店供货信息并发送到钉钉机器人,最后设置为定时任务
- crontab - 解决 mac 下通过 crontab 设置了 Python 脚本的定时任务却无法运行
- [源码解析] PyTorch分布式(5) ------ DistributedDataParallel 总述&如何使用
- Python科普系列——类与方法(上篇)
- SAP对STO的交货单执行PGI,报错 -Fld selectn for mvmt type 643 acct 400020 differs
- Spring Boot 实现通用 Auth 认证的 4 种方式
- 盘点4种使用Python批量合并同一文件夹内所有子文件夹下的Excel文件内所有Sheet数据
- OushuDB 学习经验分享(三):技术特点
- Java和Python思维方式的不同之处
- Python中日志记录新技能
- 奥比中光Gemini OpenCV—Python使用