python3 类型注释

瑜舒涵 3个月前 21浏览 0评论

Python3引入了类型注释的概念,它允许我们在代码中指定变量、函数参数和返回值的类型。类型注释能够使得代码更加清晰易懂,提升可读性、可维护性和可靠性。

def greet(name: str) -> str:
    return "Hello, " + name

在上面的代码中,我们注释了greet函数的参数类型和返回值类型。参数name的类型是一个字符串(str),返回值的类型也是一个字符串(str)。这样的注释帮助我们在编写和使用代码时更了解函数的作用和约束。

def add(a: int, b: int) -> int:
    return a + b
    
print(add(3, 4))

在上面的代码中,我们注释了add函数的两个参数和返回值的类型。要特别注意的是,这段代码中的注释并不会改变函数的行为,它只是帮助我们做好代码约束以及在IDE环境中代码提示。

我们还可以使用typing模块中的类型来注释代码。例如:

from typing import List, Tuple

def find_max(data: List[int]) -> Tuple[int, int]:
    max_index = 0
    max_value = data[0]
    for index, value in enumerate(data):
        if value > max_value:
            max_index = index
            max_value = value
    return max_index, max_value
    
print(find_max([5, 10, 2, 8, 6]))

在上面的代码中,我们使用了List和Tuple两种类型注释。List[int]表示data是一个整型数组,Tuple[int, int]表示find_max函数返回一个元组类型,其中第一个元素是最大值的索引,第二个元素是最大值本身。

使用类型注释可能会使代码变得更冗长,但它可以使得代码更加清晰易懂,便于维护、调试和扩展。因此,在编写Python3代码时,我们应该尽可能使用类型注释。