如何用类作为Python内置数值类型的类型提示？

你遇到的问题根源在于：Python内置的int、float、complex并没有显式继承numbers.Number，而是通过抽象基类的「注册机制」被纳入数值类型范畴。静态类型检查器（如mypy、Pylance）默认只认可显式继承关系，所以直接用numbers.Number会触发类型不兼容的提示。

不用Union的话，有两种可靠的替代方案：

方案1：自定义Protocol协议（推荐） #

通过typing.Protocol定义一个描述数值类型核心行为的协议，所有实现了这些行为的类型都会被类型检查器自动识别为符合要求的数值类型。

示例代码：

from typing import Protocol
from fractions import Fraction

class AnyNumber(Protocol):
    # 定义数值类型共有的核心算术方法，可根据需求增减
    def __add__(self, other: "AnyNumber") -> "AnyNumber": ...
    def __sub__(self, other: "AnyNumber") -> "AnyNumber": ...
    def __mul__(self, other: "AnyNumber") -> "AnyNumber": ...
    def __truediv__(self, other: "AnyNumber") -> "AnyNumber": ...

def foo(number: AnyNumber):
    ...

foo(1)          # 类型检查通过
foo(1.0)        # 类型检查通过
foo(1j)         # 类型检查通过
foo(Fraction()) # 类型检查通过

这种方式的优势是灵活性极强，不仅支持内置数值类型和Fraction，自定义的数值类只要实现了协议中定义的方法，也能被自动识别。

方案2：用TypeVar绑定numbers.Number并配置类型检查器 #

如果你更倾向于复用numbers.Number的定义，可以用TypeVar绑定它，同时修改类型检查器的配置，让它认可抽象基类的注册关系。

示例代码：

from typing import TypeVar
import numbers
from fractions import Fraction

# 定义一个绑定到numbers.Number的类型变量
NumberT = TypeVar("NumberT", bound=numbers.Number)

def foo(number: NumberT):
    ...

foo(1)          # 配置后类型检查通过
foo(1.0)        # 配置后类型检查通过
foo(1j)         # 配置后类型检查通过
foo(Fraction()) # 配置后类型检查通过

对应类型检查器的配置： #

• mypy：在项目根目录创建mypy.ini文件，添加以下内容：

  [mypy]
  allow_subclassing_any = True
  

• Pylance：在VS Code的设置中开启Python > Analysis: Use Library Code For Types，或者在pyrightconfig.json中添加：

  {
    "useLibraryCodeForTypes": true
  }
  

不过这种方案依赖类型检查器的配置，跨环境使用时需要确保配置同步，因此方案1的通用性更好。

内容的提问来源于stack exchange，提问作者Anonymous