预分配列表时的Python类型转换问题求教

以下是几种更优雅的解决方案：

1. 直接在初始化时使用cast #

省去中间变量，直接对初始列表做类型转换，代码更简洁：

from typing import cast, List

arr = cast(List[MyClass], [None] * n)
for i in range(n):
    # 执行计算逻辑
    arr[i] = MyClass(some_computed_value)

类型检查器会认可arr的类型为List[MyClass]，后续调用元素方法不会报错。

2. 使用assert消除Optional类型警告 #

如果坚持用Optional类型声明，可在调用元素方法前添加assert语句，明确告诉类型检查器该位置元素不为None：

from typing import Optional, List

arr: List[Optional[MyClass]] = [None] * n
for i in range(n):
    # 执行计算逻辑
    arr[i] = MyClass(some_computed_value)

# 调用方法前添加断言
assert arr[0] is not None
arr[0].my_method()

这种方式适合不想修改列表初始类型声明的场景，但需要在每个调用点前添加断言，适合元素较少的情况。

3. 改用列表推导式（如果计算逻辑允许） #

如果循环中的计算逻辑可以独立提取，直接用列表推导式生成目标列表，从根源避免预分配的类型问题：

def compute_value(i: int) -> Any:
    # 封装计算逻辑
    # ...
    return some_computed_value

arr: List[MyClass] = [MyClass(compute_value(i)) for i in range(n)]

这是最符合Python风格的写法，类型检查器无需额外提示就能识别正确类型。

4. 自定义预分配列表类（进阶复用） #

如果需要频繁使用预分配列表，可封装一个支持类型注解的容器类：

from typing import TypeVar, Generic, List, Optional

T = TypeVar('T')

class PreallocatedList(Generic[T]):
    def __init__(self, size: int):
        self._items: List[Optional[T]] = [None] * size
    
    def __setitem__(self, index: int, value: T) -> None:
        self._items[index] = value
    
    def __getitem__(self, index: int) -> T:
        item = self._items[index]
        assert item is not None, "未初始化的列表元素"
        return item
    
    def to_list(self) -> List[T]:
        assert all(item is not None for item in self._items), "所有元素必须完成初始化"
        return cast(List[T], self._items)

# 使用示例
arr = PreallocatedList[MyClass](n)
for i in range(n):
    arr[i] = MyClass(some_computed_value)
# 转换为普通列表使用
normal_arr = arr.to_list()
normal_arr[0].my_method()

这种方式封装了预分配和类型校验逻辑，适合需要重复使用该模式的场景。

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