# Python 类型提示中的类型提升-Python教程-PHP中文网

# Python 类型提示中的类型提升

花韻仙語

发布： 2025-08-12 20:00:22

原创

1037人浏览过

# Python 类型提示中的类型提升

在 Python 中，类型提示可以帮助开发者提高代码的可读性和可维护性，并允许类型检查器在运行时之前发现潜在的类型错误。然而，在某些情况下，函数的返回类型依赖于输入参数的类型，这就需要一种更高级的类型提示机制来实现类型提升。考虑一个简单的例子，一个执行乘法运算的函数，它可以接受 `int` 或 `float` 类型的参数： ```python def mul(a: int | float, b: int | float): # 返回类型？ return a * b

在这个例子中，如果 a 和 b 都是 int 类型，则函数返回 int 类型；否则，返回 float 类型。如何正确且简洁地注解这个函数的返回类型呢？

一种方法是使用 @overload 装饰器：

from typing import overload

@overload
def mul(a: int, b: int) -> int: ...

@overload
def mul(a: float, b: int | float) -> float: ...

@overload
def mul(a: int | float, b: float) -> float: ...

def mul(a, b):
    return a * b

登录后复制

这种方法虽然可行，但非常冗长，并且对于更复杂的类型组合，需要大量的重载。我们希望找到一种更简洁的方法，类似于 C++ 中的 SFINAE (Substitution failure is not an error)。

使用泛型和 TypeVar

我们可以尝试使用泛型和 TypeVar 来实现类型提升。以下是一个示例：

立即学习“Python免费学习笔记（深入）”；

from typing import TypeVar, Generic

T1 = TypeVar('T1', str, float)
T2 = TypeVar('T2', str, float)

class Promoted(Generic[T1, T2]):
    def __class_getitem__(cls, types):
        t1, t2 = types
        if t1 == str or t2 == str:
            return str
        return float

def method(a: T1, b: T2) -> Promoted[T1, T2]:
    if isinstance(a, str) or isinstance(b, str):
        return f"{a} {b}"
    else:
        return a*b

result1 = method("Hello", "World")  # 应该被推断为 str
result2 = method(3.0, 4.0)  # 应该被推断为 float
result3 = method("Price", 5.0)  # 应该被推断为 str

登录后复制

在这个例子中，我们定义了一个 Promoted 类，它是一个泛型类，可以根据 T1 和 T2 的类型来返回不同的类型。__class_getitem__ 方法用于实现类型推断的逻辑。如果 T1 或 T2 中有任何一个是 str 类型，则返回 str 类型；否则，返回 float 类型。

method 函数接受 T1 和 T2 类型的参数，并返回 Promoted[T1, T2] 类型。