什么是函數(shù)重載？簡(jiǎn)單的理解，支持多個(gè)同名函數(shù)的定義，只是參數(shù)的個(gè)數(shù)或者類型不同，在調(diào)用的時(shí)候，解釋器會(huì)根據(jù)參數(shù)的個(gè)數(shù)或者類型，調(diào)用相應(yīng)的函數(shù)。

重載這個(gè)特性在很多語(yǔ)言中都有實(shí)現(xiàn)，比如 C++、Java 等，而 Python 并不支持。這篇文章呢，通過(guò)一些小技巧，可以讓 Python 支持類似的功能。

參數(shù)個(gè)數(shù)不同的情形

先看看這種情況下 C++ 是怎么實(shí)現(xiàn)重載的

#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 按類似的方式定義函數(shù)的話，不會(huì)報(bào)錯(cuò)，只是后面的函數(shù)定義會(huì)覆蓋前面的，達(dá)不到重載的效果。

>>> 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 函數(shù)的形參十分靈活，我們可以只定義一個(gè)函數(shù)來(lái)實(shí)現(xiàn)相同的功能，就像這樣

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

參數(shù)類型不同的情形

同樣，先看下當(dāng)前情況下 C++ 的重載是怎么實(shí)現(xiàn)的

#include <iostream>
using namespace std;

int func(int a)
{
	cout << 'Int: ' << a << endl;
}

int func(float a)
{
	cout << 'Float: ' << a << endl;
}

代碼中，func 支持兩種類型的參數(shù)：整形和浮點(diǎn)型。調(diào)用時(shí)，解釋器會(huì)根據(jù)參數(shù)類型去尋找合適的函數(shù)。Python 要實(shí)現(xiàn)類似的功能，需要借助 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 函數(shù)被 functools.singledispatch 裝飾后，又根據(jù)不同的參數(shù)類型綁定了另外兩個(gè)函數(shù)。當(dāng)參數(shù)類型為整形或者浮點(diǎn)型時(shí)，調(diào)用綁定的對(duì)應(yīng)的某個(gè)函數(shù)，否則，調(diào)用自身。

執(zhí)行結(jié)果

Other: zzz
Int: 1
Float: 1.2

需要注意的是，這種方式只能夠根據(jù)第一個(gè)參數(shù)的類型去確定最后調(diào)用的函數(shù)。

關(guān)于 singledispatch 的更多細(xì)節(jié)請(qǐng)看官方文檔

https://docs.python.org/3.6/library/functools.html#functools.singledispatch

注意：函數(shù)返回值不同也是重載的一種情況，暫時(shí)沒(méi)有比較好的 Python 實(shí)現(xiàn)方式，所以沒(méi)有提及

個(gè)人覺(jué)得，重載就是為了語(yǔ)言的靈活性而設(shè)計(jì)的，而 Python 函數(shù)本來(lái)就有不少巧妙的設(shè)計(jì)，這個(gè)時(shí)候去仿這個(gè)技術(shù)，其實(shí)沒(méi)有多大必要，而且感覺(jué)有些違背 Python 的哲學(xué)。所以，本文更多的是在講如何模仿，而對(duì)于重載的使用場(chǎng)景并沒(méi)有作多少說(shuō)明。

以上所述是小編給大家介紹的Python函數(shù)重載詳解整合，希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！

最新評(píng)論