博客

python 函数如何指定参数类型？

  • Post category:Python

Python 函数可以指定参数类型，这样可以提高代码的可读性和可维护性，保证输入参数的正确性。下面我来详细讲解如何指定函数参数类型。

使用类型提示注释

在 Python 3.5 版本以后，我们可以使用类型提示注释来指定参数类型。对于函数的参数，使用 “->” 可以指定其类型，如下面的示例代码：

def greeting(name: str, age: int) -> str:
    return "Hello, {}! You are {} years old.".format(name, age)

在上述代码中，我们使用 “:” 号指定了参数的类型，其中 “str” 表示字符串类型，”int” 表示整数类型，”-> str” 表示函数返回值为字符串类型。

使用第三方库

除了使用类型提示注释外，还可以使用第三方库如 typeguard、mypy 等来进行参数类型的指定。这些库可以在函数运行之前对参数类型进行检查，并提供详细的错误信息，帮助我们快速定位错误。例如，我们可以使用 typeguard 来检查参数类型，示例代码如下：

from typeguard import typechecked

@typechecked
def greeting(name: str, age: int) -> str:
    return "Hello, {}! You are {} years old.".format(name, age)

在上述代码中，我们使用 typechecked 装饰器对函数进行装饰，这样函数运行时就会进行类型检查。

总结：

以上就是 Python 函数指定参数类型的完整攻略，其中包含了使用类型提示注释和第三方库 typeguard 进行参数类型指定的示例代码。通过指定函数参数类型，可以提高代码的可读性和可维护性，并保证输入参数的正确性。

Tags: