如何在Python中使用TypedDict为修改字典的函数添加类型提示？

当使用TypedDict定义输入（InputDict）和输出（OutputDict）类型后，直接修改输入列表副本中的字段（更改field_4类型、新增field_5）会触发类型检查报错——因为副本仍被标注为list[InputDict]，与OutputDict类型不兼容。以下是几种类型安全的解决方式：

方法一：创建新的OutputDict实例（推荐） #

这种方式通过构建全新的OutputDict字典，完全遵循类型约束，是最安全的实现方式：

from typing import TypedDict, List

class MyDict(TypedDict):
    field_1: int
    field_2: str
    field_3: float

class InputDict(MyDict):
    field_4: str

class OutputDict(MyDict):
    field_4: bytes
    field_5: bytes

def transform_dict(input: List[InputDict]) -> List[OutputDict]:
    output = []
    for item in input:
        # 展开原字典的所有字段，覆盖field_4并新增field_5
        transformed_item: OutputDict = {
            **item,
            "field_4": item["field_4"].encode(),
            "field_5": b"some_stuff"
        }
        output.append(transformed_item)
    return output

说明：通过字典解包**item复制原有字段，再显式设置符合OutputDict类型的field_4和field_5，类型检查器能准确识别每个字段的类型，完全避免类型冲突。

方法二：使用类型断言（谨慎使用） #

如果需要复用原字典对象（比如出于性能考虑），可以用cast进行类型断言，告知类型检查器我们会将字典修改为符合OutputDict的结构：

from typing import TypedDict, List, cast

class MyDict(TypedDict):
    field_1: int
    field_2: str
    field_3: float

class InputDict(MyDict):
    field_4: str

class OutputDict(MyDict):
    field_4: bytes
    field_5: bytes

def transform_dict(input: List[InputDict]) -> List[OutputDict]:
    # 将输入副本断言为OutputDict列表，允许后续修改
    output = cast(List[OutputDict], input.copy())
    for x in output:
        x["field_4"] = x["field_4"].encode()
        x["field_5"] = b"some_stuff"
    return output

注意：类型断言不会在运行时做任何检查，必须确保修改后的字典完全符合OutputDict的定义，否则可能导致运行时类型错误。

总结 #

• 优先选择方法一，它完全符合类型安全原则，代码可读性和可靠性更高；
• 仅在对性能有严格要求、且能保证修改逻辑绝对正确的场景下，再考虑使用类型断言。

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