共计 2023 个字符,预计需要花费 6 分钟才能阅读完成。
什么是函数重载?简略的了解,反对多个同名函数的定义,只是参数的个数或者类型不同,在调用的时候,解释器会依据参数的个数或者类型,调用相应的函数。
重载这个个性在很多语言中都有实现,比方 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 的哲学。所以,本文更多的是在讲如何模拟,而对于重载的应用场景并没有作多少阐明。