Python中如何为返回子类实例列表的方法添加类型提示

当前代码的问题在于：类型检查器无法通过name == 'blue'这个条件，自动推断出筛选后的实例是Blue子类，因此会抛出类型不匹配的错误。以下是几种可行的解决方案：

方案1：用isinstance做类型判断（最安全） #

直接通过isinstance判断实例类型，类型检查器会自动识别筛选结果的类型，无需额外断言：

from typing import List


class Color:
    name: str

    def __init__(self, name):
        self.name = name


class Green(Color):
    def __init__(self):
        super().__init__('green')

    def do_a_green(self):
        pass


class Blue(Color):
    def __init__(self):
        super().__init__('blue')

    def do_a_blue(self):
        pass


class ColorList:
    _color_list: List[Color]

    def __init__(self, color_list: List[Color]):
        self._color_list = color_list

    def get_blue_colors(self) -> List[Blue]:
        # 用isinstance明确判断类型，类型检查器可自动推断返回值类型
        return [color for color in self._color_list
                if isinstance(color, Blue)]

方案2：用cast做类型断言（适配特殊判断逻辑） #

如果必须通过name字段判断（而非直接检查类型），可以用typing.cast明确告知类型检查器实例的真实类型：

from typing import List, cast


class Color:
    name: str

    def __init__(self, name):
        self.name = name


class Green(Color):
    def __init__(self):
        super().__init__('green')

    def do_a_green(self):
        pass


class Blue(Color):
    def __init__(self):
        super().__init__('blue')

    def do_a_blue(self):
        pass


class ColorList:
    _color_list: List[Color]

    def __init__(self, color_list: List[Color]):
        self._color_list = color_list

    def get_blue_colors(self) -> List[Blue]:
        # 用cast强制断言类型，需确保name判断逻辑的准确性
        return [cast(Blue, color) for color in self._color_list
                if color.name == 'blue']

方案3：泛型通用筛选方法（复用性更高） #

如果需要筛选多种子类，可以用泛型实现一个通用方法，支持任意Color子类的筛选：

from typing import List, TypeVar, Type, cast

# 定义绑定到Color的泛型变量
T = TypeVar('T', bound=Color)


class Color:
    name: str

    def __init__(self, name):
        self.name = name


class Green(Color):
    def __init__(self):
        super().__init__('green')

    def do_a_green(self):
        pass


class Blue(Color):
    def __init__(self):
        super().__init__('blue')

    def do_a_blue(self):
        pass


class ColorList:
    _color_list: List[Color]

    def __init__(self, color_list: List[Color]):
        self._color_list = color_list

    def get_colors_of_type(self, color_type: Type[T]) -> List[T]:
        return [cast(T, color) for color in self._color_list
                if isinstance(color, color_type)]

# 使用示例
color_list = ColorList([Green(), Blue(), Blue()])
blues = color_list.get_colors_of_type(Blue)
for blue in blues:
    blue.do_a_blue()  # 类型检查器会自动识别blue为Blue类型，支持代码补全

方案对比 #

• isinstance方案：运行时和类型检查阶段都能保证类型正确性，是最优选择。
• cast方案：仅在类型检查阶段生效，运行时不会做类型验证，需确保判断逻辑的准确性。
• 泛型方案：一次实现即可支持所有子类筛选，适合多子类场景。

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