您的位置：首页 > 脚本大全 > > 正文

python基础编程函数参数（详解Python 函数如何重载？）

更多时间：2021-10-15 00:25:49 类别：脚本大全浏览量：1210

python基础编程函数参数

详解Python 函数如何重载？

什么是函数重载？简单的理解，支持多个同名函数的定义，只是参数的个数或者类型不同，在调用的时候，解释器会根据参数的个数或者类型，调用相应的函数。

重载这个特性在很多语言中都有实现，比如 c++、java 等，而 python 并不支持。这篇文章呢，通过一些小技巧，可以让 python 支持类似的功能。

参数个数不同的情形

先看看这种情况下 c++ 是怎么实现重载的

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17 #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 按类似的方式定义函数的话，不会报错，只是后面的函数定义会覆盖前面的，达不到重载的效果。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19 >>> 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 函数的形参十分灵活，我们可以只定义一个函数来实现相同的功能，就像这样

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18 >>> 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++ 的重载是怎么实现的

1

2

3

4

5

6

7

8

9

10

11

12 #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 装饰器。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

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