Python类型注解下isinstance判断字典值类型报错问题求解

核心结论 #

直接对字典的索引取值结果做isinstance判断，确实无法完成类型缩窄，这是类型检查工具的默认规则限制导致的。

第一个版本报错原因 #

类型检查工具（如mypy、pyright）的类型缩窄逻辑要求待判断的表达式是引用稳定的：

• 你两次访问producers[key]时，类型检查器无法保证两次取出的是同一个值——字典本身是可变结构，两次访问的间隔中，完全有可能出现其他逻辑修改了该key对应的value，因此isinstance(producers[key], list)的判断结果，无法被复用到后续对producers[key]的类型推断中，工具依然认为该值是Union[Producer, List[Producer]]类型，和返回值要求的List[Producer]不兼容，因此抛出类型错误。

临时变量写法无报错的原因 #

当你把producers[key]赋值给本地临时变量后，这个变量的引用是固定的，类型检查器默认本地变量不会被外部逻辑意外修改，因此isinstance的判断结果可以正确缩窄该变量的类型：判断为list之后，该变量的类型就会被识别为List[Producer]，赋值时自然不会报错。

代码示例 #

报错写法 #

from typing import Dict, Union, List

class Producer:
    pass

def normalize_producers(producers: Dict[str, Union[Producer, List[Producer]]]) -> Dict[str, List[Producer]]:
    res = {}
    for key in producers:
        if isinstance(producers[key], list):
            # 类型检查工具依然判定producers[key]为Union类型，报错
            res[key] = producers[key]
        else:
            res[key] = [producers[key]]
    return res

正确写法 #

def normalize_producers(producers: Dict[str, Union[Producer, List[Producer]]]) -> Dict[str, List[Producer]]:
    res = {}
    for key in producers:
        temp_val = producers[key]
        if isinstance(temp_val, list):
            res[key] = temp_val
        else:
            res[key] = [temp_val]
    return res

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