Python中多可能输入的类型提示最优方案探讨

针对你提出的场景——函数输入参数可能属于10种不同类型，以下是不同场景下的最优类型提示方案：

1. 基础方案：Union 或 Python 3.10+ 的 | 语法 #

如果这10种类型完全无关联，且该类型组合仅在少数地方使用，Union 是合法的选择，但Python 3.10及以上版本支持的竖线|语法更简洁直观，替代了冗长的Union[...]写法：

# Python 3.10+ 简洁写法
def example(input1: Type1 | Type2 | Type3 | ... | Type10) -> Type1 | Type2 | ... | Type10:
    return input1

# 兼容旧Python版本的Union写法
from typing import Union
def example(input1: Union[Type1, Type2, ..., Type10]) -> Union[Type1, Type2, ..., Type10]:
    return input1

2. 更优方案：根据类型关联选择对应策略 #

如果这10种类型存在共性，或者你需要更严谨的类型推断，以下方案会比直接用Union更好：

情况A：类型有共同父类/实现相同协议 #

如果这10种类型都继承自同一个父类，或者实现了相同的方法/属性（即符合某个协议），直接用父类或自定义Protocol作为类型提示是最优雅的选择——既简洁，又符合面向对象的依赖倒置原则：

# 示例：所有类型都继承自BaseClass
def example(input1: BaseClass) -> BaseClass:
    return input1

# 示例：自定义Protocol（适用于无继承关系但有共同接口的类型）
from typing import Protocol

class HasCommonMethod(Protocol):
    def common_method(self) -> None:
        ...

# 只要类型实现了common_method，就符合该Protocol
def example(input1: HasCommonMethod) -> HasCommonMethod:
    return input1

情况B：需要保留输入输出的类型一致性 #

如果你的函数逻辑是返回与输入相同类型的对象（比如示例中的return input1），使用有界TypeVar能让类型检查器精准推断输入输出的具体类型，而不是返回一个模糊的Union类型：

from typing import TypeVar

# 绑定允许的10种类型
InputType = TypeVar("InputType", Type1, Type2, ..., Type10)

def example(input1: InputType) -> InputType:
    return input1

此时，当你传入Type1的实例，类型检查器会明确知道返回值也是Type1，而非所有类型的联合。

情况C：多场景复用该类型组合 #

如果这10种类型的组合在代码中多次出现，用TypeAlias定义一个别名，能大幅提升代码可读性和可维护性：

from typing import TypeAlias

# Python 3.10+ 直接用|定义别名
MyAllowedTypes: TypeAlias = Type1 | Type2 | ... | Type10

def example(input1: MyAllowedTypes) -> MyAllowedTypes:
    return input1

总结 #

没有绝对的“最佳方案”，需根据实际场景选择：

• 无关联类型+少次使用：优先用Python 3.10+的|语法
• 需保留类型一致性：用有界TypeVar
• 类型有共同接口：用父类或Protocol
• 多场景复用类型组合：自定义TypeAlias

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