如何重载 Python 函数?

pythonserver side programmingprogramming

在 Python 中,您可以定义一个方法,以便有多种调用方式。根据函数定义,可以使用零个、一个、两个或更多个参数来调用它。这称为方法重载。

在给定的代码中,有一个类,只有一个方法 sayHello()。我们重写如下。此方法的第一个参数设置为 None,这使我们可以选择使用或不使用参数来调用它。

基于类创建一个对象,我们使用零个和一个参数来调用其方法。为了实现方法重载,我们以两种方式调用方法 sayHello():1. obj.sayHello() 和 2. obj.sayHello('Rambo')

我们创建了一个方法,可以使用比定义允许的参数更少的参数来调用。我们不仅限于两个变量,给定的方法可以有更多可选变量。

示例

class Human:
   def sayHello(self, name=None): 
      if name is not None:
         print ('Hello ' + name)
      else:
         print ('Hello ') 
obj = Human() 
print(obj.sayHello())
print(obj.sayHello('Rambo'))

输出

Hello 
None
Hello Rambo
None

在 Python 中,函数重载并非原生支持,而在其他编程语言(如 C++)中则支持。但是,我们可以通过使用默认参数和/或可变长度参数来实现相同的效果。

示例

下面是一个使用默认参数的示例

在此示例中,函数 add() 可以接受两个或三个参数。如果只传递一个参数,它将被赋值 a,其他两个参数将采用其默认值 0。如果传递了两个参数,它们将被赋值给 a 和 b,而 c 将采用其默认值。如果传递了所有三个参数,它们将分别被赋值给 a、b 和 c。

def add(a, b=0, c=0): 
   return a + b + c
print(add(2))

输出

上述代码产生以下输出

2

示例

def add(a, b=0, c=0): 
   return a + b + c
print(add(3,2))

输出

上述代码产生以下输出

5

示例

def add(a, b=0, c=0):
   return a + b + c
print(add(1,4,3))

输出

上述代码产生以下输出

8

示例

在 Python 中实现函数重载的另一种方法是使用可变长度参数。以下是示例 -

def add(*args):
   result = 0
   for arg in args:
      result += arg
   return result
print(add(1,2,3))

输出

上述代码产生以下输出

6

示例

def add(*args):
   result = 0
   for arg in args:
      result += arg
   return result 
print(add(1,3,4,5))

输出

上述代码产生以下输出

13

在此示例中,函数 add 接受可变数量的参数。所有参数都打包到名为 args 的元组中,函数将元组中的所有值相加并返回结果。

通过使用默认参数和可变长度参数,我们可以创建根据传递给它们的参数数量和类型而表现不同的函数,这是函数重载的本质。

示例

使用不同的参数类型重载函数。

在此示例中,函数 multiply 被重载以接受两种不同类型的参数:int 和 float。函数检查参数的类型,并且仅当它们属于同一类型时才执行乘法。如果类型不同,函数将返回错误消息。

def multiply(a, b):
   if isinstance(a, int) and isinstance(b, int):
      return a * b
   elif isinstance(a, float) and isinstance(b, float): 
      return a * b
   else:
      return "Invalid argument types" 
print(multiply(2, 3))

输出

上述代码产生以下输出

6

示例

def multiply(a, b):
    if isinstance(a, int) and isinstance(b, int): 
       return a * b
    elif isinstance(a, float) and isinstance(b, float):
       return a * b
    else:
       return "Invalid argument types"
print(multiply(2.5, 3.5))

输出

上述代码产生以下输出

8.75

示例

def multiply(a, b):
   if isinstance(a, int) and isinstance(b, int): 
      return a * b
   elif isinstance(a, float) and isinstance(b, float):
      return a * b
   else:
       return "Invalid argument types"
print(multiply(2, 3.5))

输出

上述代码产生以下输出

Invalid argument types

示例

使用不同数量的参数重载函数

在此示例中,函数 add 使用两个不同数量的参数重载。但是,Python 不允许仅基于参数数量进行函数重载,并且 add 的第二个定义将覆盖第一个定义。因此,当我们尝试仅使用两个参数调用 add 时,我们会得到 TypeError,因为该函数需要三个参数。

def add(a, b): 
   return a + b
def add(a, b, c): 
   return a + b + c
print(add(2, 3))

输出

上述代码产生以下输出

Traceback (most recent call last):
  File "/home/cg/root/42648/main.py", line 5, in 
    print(add(2, 3))
TypeError: add() missing 1 required positional argument: 'c'

示例

def add(a, b): 
   return a + b

def add(a, b, c): 
   return a + b + c
print(add(2,3,4))

输出

上述代码产生以下输出

9

因此,在本文中,我们看到了在 Python 中实现函数重载的不同方法。


相关文章