如何为自定义List类的reduce方法添加正确的类型注解？

你的核心问题出在类型注解的错误定义和泛型类的缺失声明上，同时reduce方法的返回逻辑也可能不符合预期：

1. 错误原因拆解 #

• func的类型注解完全错误：原代码中把func标注为Callable[[list[T], list[T]], list[T]]，但functools.reduce处理self.list（即list[T]）时，会将序列中的T类型元素依次传入func，而非list[T]类型。比如字符串列表场景下，func接收的是两个str，而非list[str]，这直接导致类型检查器报错。
• 封装类未声明为泛型类：你的List类没有继承Generic[T]，类型变量T无法被类型检查器正确推导，进一步加剧了类型错误。
• 返回逻辑可能不符合预期：原代码中List(reduce(func, self.list))会把reduce的结果（比如字符串'ab'）当作可迭代对象拆解，最终得到List(['a','b'])，而非你可能预期的List(['ab'])。

2. 修正后的完整实现 #

from typing import Generic, TypeVar, Callable, Optional, overload
from functools import reduce

T = TypeVar('T')
U = TypeVar('U')

class List(Generic[T]):
    def __init__(self, lst: list[T]) -> None:
        self.list = lst

    @overload
    def reduce(self, func: Callable[[T, T], T]) -> 'List[T]':
        """无默认值时，累积类型与元素类型一致"""
        ...

    @overload
    def reduce(self, func: Callable[[U, T], U], default: U) -> 'List[U]':
        """有默认值时，累积类型可以是任意类型U"""
        ...

    def reduce(self, func: Callable[..., object], default: Optional[object] = None) -> 'List[object]':
        if default is None:
            result = reduce(func, self.list)
        else:
            result = reduce(func, self.list, default)
        # 若要将累积结果作为单个元素存入List，用List([result])
        # 若要将结果拆解为可迭代对象（比如字符串拆成字符），用List(list(result))
        return List([result])

3. 测试验证 #

# 字符串列表场景
a: List[str] = List(['a', 'b'])
b = a.reduce(lambda x, y: x + y)
# b的类型为List[str]，值为List(['ab'])，类型检查器无报错

# 带默认值的场景
c: List[int] = List([1, 2, 3])
d = c.reduce(lambda acc, num: acc + num, 0)
# d的类型为List[int]，值为List([6])

关键说明 #

• 使用@overload区分有无默认值的两种情况，让类型检查器能正确推导返回类型。
• 泛型类List(Generic[T])确保类型变量T在整个类中能被正确识别。
• 根据实际需求调整返回逻辑：如果需要将累积结果作为单个元素存储，用List([result])；如果需要拆解结果（比如字符串转字符列表），用List(list(result))。

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