Python的高级用法：泛型
在 Python 中，泛型（Generic Programming） 是一种编程范式，它允许我们编写能够处理多种数据类型的代码，而不需要为每种类型单独实现代码。这种方法提高了代码的复用性和灵活性，是高级 Python 编程中不可或缺的一部分。
本文将介绍泛型在 Python 中的概念、用法，以及如何通过泛型提升代码的灵活性，结合代码示例和图解，让你轻松掌握这一高级特性。
一、泛型的概念
泛型编程的核心思想是类型的参数化。通过类型参数化，我们可以在编译时或运行时指定类型，而不是在编写代码时硬编码某种特定的类型。
在 Python 中，泛型主要体现在以下场景：
- 函数和类的类型注解
- 标准库中的泛型容器（如 List、Dict、Set 等）
- 自定义泛型类型
二、泛型的基础用法
1. 泛型类型注解
在 Python 中，可以使用 typing 模块来实现泛型注解。例如，指定一个列表只能包含整数或字符串：
from typing import List
def sum_elements(elements: List[int]) -> int:
return sum(elements)
print(sum_elements([1, 2, 3])) # 输出：6
# print(sum_elements(["a", "b", "c"])) # 报错：类型检查工具会警告泛型改进
如果函数需要接受多种类型的列表，比如整数或浮点数，可以通过泛型类型来实现：
from typing import TypeVar, List
T = TypeVar('T', int, float)
def sum_elements(elements: List[T]) -> T:
return sum(elements)
print(sum_elements([1, 2, 3])) # 输出：6
print(sum_elements([1.5, 2.5, 3])) # 输出：7.02. 泛型类
泛型不仅适用于函数，也适用于类。通过 Generic 类，可以定义参数化的类。
from typing import Generic, TypeVar
T = TypeVar('T')
class Box(Generic[T]):
def __init__(self, item: T):
self.item = item
def get_item(self) -> T:
return self.item
# 使用 Box 保存不同类型的对象
int_box = Box(123)
str_box = Box("Hello, Generic!")
print(int_box.get_item()) # 输出：123
print(str_box.get_item()) # 输出：Hello, Generic!三、应用场景
1. 类型安全的容器
泛型容器可以确保只有特定类型的数据能够存储在容器中。例如：
from typing import List, TypeVar
T = TypeVar('T')
class Stack(Generic[T]):
def __init__(self):
self.items: List[T] = []
def push(self, item: T):
self.items.append(item)
def pop(self) -> T:
return self.items.pop()
# 创建一个只接受整数的栈
int_stack = Stack[int]()
int_stack.push(1)
int_stack.push(2)
print(int_stack.pop()) # 输出：2
# 创建一个只接受字符串的栈
str_stack = Stack[str]()
str_stack.push("Hello")
str_stack.push("World")
print(str_stack.pop()) # 输出：World2. 函数的多类型支持
通过泛型函数，可以让函数接受不同类型的输入：
from typing import Union, TypeVar
T = TypeVar('T', int, str)
def repeat(item: T, times: int) -> List[T]:
return [item] * times
print(repeat(5, 3)) # 输出：[5, 5, 5]
print(repeat("Hi", 2)) # 输出：['Hi', 'Hi']3. 数据处理工具
泛型适合构建灵活的数据处理工具，比如过滤、映射等操作：
from typing import Callable, List, TypeVar
T = TypeVar('T')
def filter_items(items: List[T], predicate: Callable[[T], bool]) -> List[T]:
return [item for item in items if predicate(item)]
numbers = [1, 2, 3, 4, 5]
print(filter_items(numbers, lambda x: x > 3)) # 输出：[4, 5]四、类型推断与运行时检查
1. 类型推断
Python 中的类型注解是静态的，主要用于开发阶段的类型检查工具（如 mypy）：
from typing import List
def double_numbers(numbers: List[int]) -> List[int]:
return [x * 2 for x in numbers]
# mypy 会检查类型是否匹配
print(double_numbers([1, 2, 3])) # 输出：[2, 4, 6]2. 运行时的类型检查
Python 运行时不会强制类型检查，但可以通过 isinstance 检查类型：
def process_items(items: List[int]):
for item in items:
if not isinstance(item, int):
raise ValueError("All items must be integers")
print(item)
process_items([1, 2, 3]) # 正常
# process_items([1, "a", 3]) # 抛出 ValueError五、图解泛型
下图展示了泛型函数和类的工作流程：
泛型函数 泛型类
↓ ↓
输入多种类型 生成实例化对象
↓ ↓
运行时参数化 进行类型推断
↓ ↓
返回泛型结果 提供类型安全的操作六、注意事项
运行时无效
- 泛型注解只在开发阶段有效，运行时并不会强制类型检查。
过度使用可能导致复杂性
- 在简单项目中，避免过度泛型化，可能会让代码难以理解。
与协变、逆变的关系
- 泛型支持协变和逆变，可以更灵活地控制子类和父类之间的类型关系。
七、总结
通过泛型，我们可以编写更具通用性和可维护性的代码，提高代码复用率。无论是构建类型安全的容器，还是开发灵活的数据处理工具，泛型在 Python 编程中都有着广泛的应用场景。
延伸阅读
用泛型简化代码，为你的 Python 项目增添更多灵活性！