Python List/Dict/Set：容器类型注解用法详解

01-24 4049阅读

在Python编程中，List、Dict和Set是非常常用的容器类型。它们各自有着独特的用途和特点，而类型注解则能让代码更加清晰易懂，有助于提高代码的可读性和可维护性。

List类型注解

List是Python中用来存储有序序列的容器。通过类型注解，可以明确列表中元素的类型。

from typing import List

numbers: List[int] = [1, 2, 3, 4, 5]
words: List[str] = ["apple", "banana", "cherry"]

def sum_list(lst: List[int]) -> int:
    total = 0
    for num in lst:
        total += num
    return total

print(sum_list(numbers))

在上述代码中，numbers被注解为List[int]，表示该列表中的元素都是整数类型。words被注解为List[str]，表示其元素为字符串类型。sum_list函数接受一个整数列表作为参数，并返回列表元素的总和。类型注解使得函数的参数和返回值类型一目了然，避免了潜在的类型错误。

Python List/Dict/Set：容器类型注解用法详解

Dict类型注解

Dict用于存储键值对的无序集合。类型注解可以指定键和值的类型。

from typing import Dict

person: Dict[str, int] = {"age": 30, "height": 175}
product: Dict[str, float] = {"price": 9.99, "discount": 0.1}

def get_value(dct: Dict[str, int], key: str) -> int:
    return dct[key]

print(get_value(person, "age"))

这里，person被注解为Dict[str, int]，表示键是字符串类型，值是整数类型。product被注解为Dict[str, float]，键为字符串，值为浮点数。get_value函数接受一个字典和一个键作为参数，返回对应键的值，通过类型注解确保了函数操作的正确性。

Set类型注解

Set是无序且元素唯一的集合。类型注解能明确集合中元素的类型。

from typing import Set

numbers_set: Set[int] = {1, 2, 3, 3, 4}
letters_set: Set[str] = {"a", "b", "b", "c"}

def contains_number(s: Set[int], num: int) -> bool:
    return num in s

print(contains_number(numbers_set, 2))

numbers_set被注解为Set[int]，集合中的元素都是整数，且不会有重复。letters_set被注解为Set[str]。contains_number函数用于检查一个集合中是否包含某个特定的整数，类型注解保证了函数参数和返回值的类型一致性。