typing库

最新推荐文章于 2024-08-03 13:59:41 发布

原创最新推荐文章于 2024-08-03 13:59:41 发布

· 361 阅读

3 ·

版权

文章标签：

#windows

typing 库是 Python 标准库的一部分，用于支持静态类型检查。通过使用 typing 库，你可以为函数、变量和类添加类型注解，从而提高代码的可读性和可维护性，并帮助工具如 IDE 和类型检查器（例如 mypy）进行静态类型检查。

以下是 typing 库的一些常见用法和示例：

1. 基本类型注解

你可以为函数参数和返回值添加类型注解：

def greet(name: str) -> str:
    return f"Hello, {name}!"

print(greet("Alice"))

2. 使用 `List` 和 `Dict`

List 和 Dict 用于注解列表和字典的类型：

from typing import List, Dict

def process_numbers(numbers: List[int]) -> int:
    return sum(numbers)

def process_dict(data: Dict[str, int]) -> int:
    return sum(data.values())

print(process_numbers([1, 2, 3]))
print(process_dict({"a": 1, "b": 2, "c": 3}))

3. 使用 `Tuple`

Tuple 用于注解元组的类型：

from typing import Tuple

def get_coordinates() -> Tuple[int, int]:
    return (10, 20)

print(get_coordinates())

4. 使用 `Optional`

Optional 用于表示某个值可以是某种类型或者是 None：

from typing import Optional

def find_item(items: List[str], item: str) -> Optional[int]:
    try:
        return items.index(item)
    except ValueError:
        return None

print(find_item(["apple", "banana", "cherry"], "banana"))
print(find_item(["apple", "banana", "cherry"], "orange"))

5. 使用 `Union`

Union 用于表示某个值可以是多种类型中的一种：

from typing import Union

def process_value(value: Union[int, str]) -> str:
    if isinstance(value, int):
        return f"Integer: {value}"
    else:
        return f"String: {value}"

print(process_value(10))
print(process_value("hello"))

6. 使用 `Any`

Any 用于表示任意类型：

from typing import Any

def process_anything(value: Any) -> str:
    return f"Received: {value}"

print(process_anything(10))
print(process_anything("hello"))
print(process_anything([1, 2, 3]))

7. 使用 `Callable`

Callable 用于表示可调用对象（如函数）的类型：

from typing import Callable

def execute_function(func: Callable[[int, int], int], a: int, b: int) -> int:
    return func(a, b)

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

print(execute_function(add, 2, 3))

8. 使用 `TypeVar`

TypeVar 用于定义泛型类型：

from typing import TypeVar, List

T = TypeVar('T')

def get_first_element(elements: List[T]) -> T:
    return elements[0]

print(get_first_element([1, 2, 3]))
print(get_first_element(["a", "b", "c"]))

9. 使用 `Generic`

Generic 用于创建泛型类：

from typing import TypeVar, Generic

T = TypeVar('T')

class Box(Generic[T]):
    def __init__(self, content: T):
        self.content = content

    def get_content(self) -> T:
        return self.content

int_box = Box(123)
str_box = Box("hello")

print(int_box.get_content())
print(str_box.get_content())

10. 使用 `Literal`

Literal 用于表示具体的值类型：

from typing import Literal

def process_status(status: Literal["open", "closed"]) -> str:
    if status == "open":
        return "The case is open."
    else:
        return "The case is closed."

print(process_status("open"))
print(process_status("closed"))