Python泛型编程：TypeVar和Generic详解 - 写给初学者的指南

原创已于 2024-12-11 09:47:47 修改 · 1.3k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #TypeVar #Generic

于 2024-12-10 11:17:30 首次发布

LINUX 专栏收录该内容

945 篇文章

订阅专栏

Python泛型编程：TypeVar和Generic详解 - 写给初学者的指南

前言

大家好！今天我们来聊一聊Python中的泛型编程。想象一下，泛型就像是一个"万能盒子"，可以装不同类型的东西，但又保持它的灵活性和类型安全。听起来很酷，对吧？

1. 为什么需要泛型？

在编程中，我们经常需要编写可以处理多种类型数据的函数或类。比如，你希望创建一个函数，无论是处理整数、字符串还是自定义对象，都能正常工作。这就是泛型发挥作用的地方！

2. TypeVar：定义泛型类型变量

TypeVar就像是一个类型的"通行证"，它允许你定义可以接受不同类型的变量。

from typing import TypeVar, Generic

# 定义一个泛型变量T
T = TypeVar('T')

# 创建一个泛型函数
def get_first_item(items: list[T]) -> T:
    """获取列表的第一个元素"""
    if items:
        return items[0]
    raise ValueError("列表为空")

# 使用示例
numbers = [1, 2, 3, 4, 5]
words = ['apple', 'banana', 'cherry']

print(get_first_item(numbers))  # 输出: 1
print(get_first_item(words))    # 输出: apple

3. Generic：创建泛型类

我们可以使用Generic创建支持多种类型的类。想象一下，我们要创建一个可以存储任何类型数据的"魔法盒子"。

from typing import TypeVar, Generic

# 定义泛型变量
T = TypeVar('T')

class MagicBox(Generic[T]):
    def __init__(self):
        self._items = []

    def add_item(self, item: T):
        """添加物品到盒子"""
        self._items.append(item)

    def get_items(self) -> list[T]:
        """获取盒子中的所有物品"""
        return self._items

    def get_first_item(self) -> T:
        """获取第一个物品"""
        if self._items:
            return self._items[0]
        raise ValueError("盒子是空的")

# 创建不同类型的魔法盒子
number_box = MagicBox[int]()
number_box.add_item(10)
number_box.add_item(20)
print(number_box.get_items())  # 输出: [10, 20]

string_box = MagicBox[str]()
string_box.add_item("Hello")
string_box.add_item("World")
print(string_box.get_items())  # 输出: ['Hello', 'World']

4. 多个泛型类型变量

我们还可以使用多个泛型类型变量，就像有多个"通行证"。

from typing import TypeVar, Generic

K = TypeVar('K')  # Key类型
V = TypeVar('V')  # Value类型

class SuperDict(Generic[K, V]):
    def __init__(self):
        self._data = {}

    def add_pair(self, key: K, value: V):
        """添加键值对"""
        self._data[key] = value

    def get_value(self, key: K) -> V:
        """根据键获取值"""
        return self._data.get(key)

# 创建不同类型的超级字典
name_age_dict = SuperDict[str, int]()
name_age_dict.add_pair("Alice", 25)
name_age_dict.add_pair("Bob", 30)
print(name_age_dict.get_value("Alice"))  # 输出: 25

code_language_dict = SuperDict[int, str]()
code_language_dict.add_pair(1, "Python")
code_language_dict.add_pair(2, "Java")
print(code_language_dict.get_value(1))  # 输出: Python