Python类型提示：让泛型类型推导在继承场景下生效

你的问题出在MapNode的泛型类型变量定义上，当前的New_IT和New_OT与T_Lambda_In、T_Lambda_Out之间没有建立清晰的关联，导致类型检查器无法从lambda函数的签名和输入列表推导出具体的类型。加上Python 3.6的typing模块对泛型推导的支持相对有限，需要调整类型变量的结构来让推导生效。

调整后的实现方案 #

我们可以直接让MapNode的泛型参数与lambda的输入输出类型绑定，去掉多余的New_IT和New_OT，这样类型检查器就能明确关联列表元素类型和lambda的参数/返回值类型：

from typing import TypeVar, Generic, Callable, List

T_In = TypeVar("T_In")
T_Out = TypeVar("T_Out")

class Node(Generic[T_In, T_Out]):
    input_: T_In
    output: T_Out

    def __init__(self, input_: T_In):
        self.input_ = input_
        self.output = None

    def prepare_output(self):
        raise NotImplementedError()

    def get_output(self) -> T_Out:
        return self.output

# 重新定义MapNode的类型变量，直接关联lambda的输入输出
T_Lambda_In = TypeVar("T_Lambda_In")
T_Lambda_Out = TypeVar("T_Lambda_Out")

class MapNode(Node[List[T_Lambda_In], List[T_Lambda_Out]]):
    def __init__(self, input_: List[T_Lambda_In], lambda_: Callable[[T_Lambda_In], T_Lambda_Out]):
        super().__init__(input_)
        self.lambda_ = lambda_

    def prepare_output(self):
        self.output = [self.lambda_(obj) for obj in self.input_]

# 示例转换函数
def cast_int(val: str) -> int:
    return int(val)

# 现在类型推导会正常工作
node = MapNode(["1", "2"], lambda_=cast_int)
# PyCharm会正确识别node的类型为MapNode[List[str], List[int]]

关键调整说明 #

1. 简化泛型参数：把MapNode直接继承自Node[List[T_Lambda_In], List[T_Lambda_Out]]，这样T_Lambda_In和T_Lambda_Out直接对应输入列表的元素类型和输出列表的元素类型，类型检查器可以通过输入列表和lambda的签名直接推导这两个类型变量。
2. 明确输入参数类型：__init__中的input_参数明确标注为List[T_Lambda_In]，让类型检查器能从传入的["1", "2"]推断出T_Lambda_In是str，再通过cast_int的签名推断出T_Lambda_Out是int。
3. 修复super调用：你原来的super(Node, self).__init__(input_)写错了，应该是super().__init__(input_)或者super(MapNode, self).__init__(input_)，否则会跳过Node类的初始化逻辑，这也是潜在的bug。

验证效果 #

调整后，PyCharm会正确推导出node的类型为MapNode[List[str], List[int]]，符合你的预期。如果在Python 3.6中仍然有轻微的推导问题，可以尝试显式指定泛型参数（虽然通常不需要）：

node: MapNode[List[str], List[int]] = MapNode(["1", "2"], lambda_=cast_int)

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