如何为函数/方法参数添加泛型类型注解以支持任意实数类型？

核心问题分析 #

numbers.Real ABC在运行时能通过isinstance正确识别所有内置实数类型，但主流类型检查器会忽略ABC.register注册的虚拟子类，因此无法直接用它做可靠的类型注解。官方没有现成的SupportsReal协议，这时候可以通过自定义结构化子类型协议来解决。

方案1：自定义核心实数运算协议 #

你可以自己定义SupportsReal协议，不需要枚举所有实数运算方法，只需要覆盖你的函数实际用到的核心方法即可——类型检查器会基于这些方法验证类型兼容性。

示例代码：

from typing import Protocol, TypeVar

T = TypeVar('T', bound='SupportsReal')

class SupportsReal(Protocol):
    def __add__(self: T, other: T) -> T: ...
    def __sub__(self: T, other: T) -> T: ...
    def __mul__(self: T, other: T) -> T: ...
    def __truediv__(self: T, other: T) -> T: ...
    def __neg__(self) -> T: ...
    def __abs__(self) -> T: ...
    # 按需添加其他需要的方法，比如__floordiv__、__pow__等

def my_real_calc(a: T, b: T) -> T:
    return (a + b) * (a - b)

这样定义后，int、float、fractions.Fraction以及任何实现了这些方法的第三方实数类型，都会被类型检查器识别为符合协议的类型。

方案2：兼顾运行时检查与类型注解 #

如果需要同时保证运行时类型安全和类型检查有效性，可以在注解中用自定义协议，运行时仍用numbers.Real做检查：

示例代码：

from numbers import Real
from typing import Protocol, TypeVar

T = TypeVar('T', bound='SupportsReal')

class SupportsReal(Protocol):
    # 定义核心方法...
    def __add__(self: T, other: T) -> T: ...
    def __truediv__(self: T, other: T) -> T: ...

def process_real(a: T, b: T) -> T:
    # 运行时校验
    if not isinstance(a, Real) or not isinstance(b, Real):
        raise TypeError("参数必须是实数类型")
    return a / b + a * b

为什么不用numbers.Real直接做注解？ #

主流类型检查器（如mypy）只会识别显式继承numbers.Real的类，而Python内置的int、float等都是通过虚拟注册关联到numbers.Real的，因此类型检查器不会将它们判定为numbers.Real的子类，导致注解失效。

注意事项 #

• 协议无需过度臃肿，只定义函数实际用到的方法即可，避免不必要的类型校验。
• 第三方类型若未实现协议中的方法，类型检查器会及时抛出提示，提前发现兼容性问题。

内容的提问来源于stack exchange，提问作者I Like Python