如何为支持比较操作的参数添加Python Type hint？

针对你需要为支持大于/小于比较操作的参数添加类型提示的需求，有几种可行的实现方式：

1. 自定义Protocol（推荐，Python 3.8+） #

利用typing.Protocol定义一个代表"可比较"的协议类型，静态类型检查器（如mypy、pyright）会自动识别所有实现了比较方法的对象符合该类型。

from typing import Protocol, TypeVar

class Comparable(Protocol):
    def __lt__(self, other: "Comparable") -> bool: ...
    def __le__(self, other: "Comparable") -> bool: ...
    def __gt__(self, other: "Comparable") -> bool: ...
    def __ge__(self, other: "Comparable") -> bool: ...

# 定义受约束的类型变量，确保参数类型一致
T = TypeVar('T', bound=Comparable)

def trim_dataset(data: list[T], lower_bound: T, upper_bound: T) -> list[T]:
    """修剪有序数据集，保留在上下界之间的元素"""
    return [item for item in data if lower_bound <= item <= upper_bound]

这个方案灵活性极强：int、float、datetime.datetime等内置类型，以及你自己实现了对应比较方法的自定义类，都会被静态检查器判定为符合Comparable类型。

如果需要兼容Python 3.7及更早版本，可以使用typing_extensions.Protocol代替标准库的Protocol。

2. 直接绑定常用类型（适合明确限定范围的场景） #

如果你的函数只需要支持特定的几种可比较类型（比如仅int、float和时间戳），可以直接用Union组合这些类型：

from datetime import datetime
from typing import Union

ComparableType = Union[int, float, datetime]

def trim_dataset(data: list[ComparableType], lower_bound: ComparableType, upper_bound: ComparableType) -> list[ComparableType]:
    return [item for item in data if lower_bound <= item <= upper_bound]

这种方式更直观，但缺点是无法自动兼容自定义的可比较类，扩展性较差。

3. 使用TypeVar结合单操作协议 #

如果只需要保证参数支持特定的比较操作（比如仅小于/大于），可以使用标准库中提供的SupportsLessThan、SupportsGreaterThan等协议：

from typing import TypeVar, SupportsLessThan, SupportsGreaterThan

T = TypeVar('T', bound=SupportsLessThan[T] & SupportsGreaterThan[T])

def trim_dataset(data: list[T], lower_bound: T, upper_bound: T) -> list[T]:
    return [item for item in data if lower_bound <= item <= upper_bound]

不过这种方式需要同时组合多个单操作协议，不如自定义完整的Comparable协议简洁。

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