如何定义具有特定键值类型对应关系的Python字典类型提示？

你遇到的问题根源在于：用Union[Dict[Fruit, int], Dict[Car, str]]声明的变量，类型检查器无法确定它具体是哪一种字典实例，会同时对两种可能的分支做校验——既担心它是Dict[Car, str]（此时用Fruit当键不合法），又认为返回值可能是int|str（无法直接赋值给int）。

下面是几种可行的解决方式：

1. 函数重载（推荐用于函数参数/返回值） #

如果这个字典是函数的返回值或参数，用@overload明确不同的类型分支，让类型检查器能根据上下文推断具体类型：

from typing import overload, Dict
from dataclasses import dataclass

@dataclass
class Fruit:
    id: int

@dataclass
class Car:
    model: str

# 先定义重载签名
@overload
def get_target_dict() -> Dict[Fruit, int]: ...
@overload
def get_target_dict() -> Dict[Car, str]: ...

# 实际实现
def get_target_dict():
    # 这里可以根据逻辑返回其中一种字典
    return {Fruit(0): 100}

# 类型检查器能推断出test是Dict[Fruit, int]
test = get_target_dict()
number: int = test[Fruit(0)]  # 无类型错误

2. 类型断言（适合明确变量类型的场景） #

如果你确定当前变量的具体类型，可以用类型断言直接告诉类型检查器，跳过不确定的分支校验：

from typing import Dict, Union

test: Union[Dict[Fruit, int], Dict[Car, str]] = {Fruit(0): 100}
# 用类型断言指定当前test是Dict[Fruit, int]
number: int = (test as Dict[Fruit, int])[Fruit(0)]  # 类型检查通过

注意：这种方式需要你自己保证实际类型和断言一致，否则会引发运行时错误。

3. 用Protocol定义专属字典协议 #

通过Protocol定义两种符合要求的字典行为，替代直接用Dict的Union，让类型检查器能精准识别键和值的类型：

from typing import Protocol, Union

class FruitDict(Protocol):
    def __getitem__(self, key: Fruit) -> int: ...
    # 可以按需添加其他字典方法，比如__contains__、__setitem__等

class CarDict(Protocol):
    def __getitem__(self, key: Car) -> str: ...

test: Union[FruitDict, CarDict] = {Fruit(0): 100}
number: int = test[Fruit(0)]  # 类型检查通过

这种方式更灵活，哪怕不是标准dict实例，只要符合协议的类都能被正确识别。

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