Python 函数参数指定类型|极客教程

Python 函数参数指定类型

在 Python 中，函数参数的类型是动态的，这意味着你可以在不指定类型的情况下传递各种不同类型的参数。然而，有时候我们希望函数可以接受特定类型的参数，这可以帮助我们更好地控制和理解函数的行为。从 Python 3.5 开始，我们可以使用类型提示和注解来指定函数参数的类型，让代码更加清晰和易于理解。

1. 类型提示和注解

类型提示是在函数定义或者函数参数前面加上类型信息，注解则是使用 -> 指定函数的返回值类型。这两者可以一起使用，让我们的代码更具可读性。

下面是一个简单的示例，展示了如何在函数参数和返回值中使用类型提示和注解：

def add(x: int, y: int) -> int:
    return x + y

result = add(3, 5)
print(result)  # Output: 8

在这个示例中，我们定义了一个名为 add 的函数，它接受两个整数类型的参数 x 和 y，并且返回一个整数类型的值。通过类型提示和注解，我们清晰地表明了这个函数所期望的参数类型和返回值类型。

2. 函数参数类型注解

在函数定义中，你可以给参数加上类型提示，这样当调用函数时传入不符合类型的参数时，Python 解释器会在运行时检查参数的类型是否匹配。这可以帮助减少程序运行时可能出现的类型错误。

让我们看一个示例：

def multiply(x: int, y: int) -> int:
    return x * y

result = multiply(4, "hello")

在这个示例中，我们定义了一个乘法函数 multiply，期望两个整数类型的参数。然而，我们却传入了一个整数和一个字符串，这将导致 TypeError: can't multiply sequence by non-int of type 'int' 的错误。

3. 函数返回值类型注解

除了参数类型外，我们还可以注解函数的返回值类型。这有助于让其他开发人员更容易理解函数的功能和预期输出。

下面是一个示例：

def divide(x: int, y: int) -> float:
    return x / y

result = divide(10, 3)

在这个示例中，我们定义了一个除法函数 divide，接受两个整数类型的参数并返回一个浮点数类型的值。

4. 可选参数和默认值

在定义函数参数时，我们可以使用 Optional 和 Union 类型标注来指定可选参数和多种可能的参数类型，同时也可以设置默认值。

下面是一个示例：

from typing import Optional, Union

def greet(name: str, age: Optional[int] = None) -> str:
    if age:
        return f"Hello, {name}! You are {age} years old."
    else:
        return f"Hello, {name}!"

result1 = greet("Alice")
result2 = greet("Bob", 25)

print(result1)  # Output: Hello, Alice!
print(result2)  # Output: Hello, Bob! You are 25 years old.

在这个示例中，我们定义了一个 greet 函数，它接受一个字符串类型的 name 参数和一个可选的整数类型的 age 参数。如果 age 被传入，函数会返回一个带有年龄信息的问候语，否则只返回简单的问候语。