当然可以实现！你想要的就是让Python的静态类型检查器能根据传入的TypedDict键，自动推断函数返回值的类型，类似TypeScript里keyof泛型约束的效果。下面给你一步步拆解实现方法：

首先，我们需要用到Python typing模块里的几个工具：TypedDict、Literal、TypeVar。先看完整的可运行代码：

from typing import TypedDict, Literal, TypeVar

class MyType(TypedDict):
    x: int
    y: str

# 定义一个受约束的类型变量，只能是MyType的合法键
TKey = TypeVar('TKey', bound=Literal['x', 'y'])

def func(t: MyType, k: TKey) -> MyType[TKey]:
    # 这里可以添加你需要的业务逻辑
    return t[k]

# 测试使用
a = func({'x': 1, 'y': '2'}, 'x')  # 类型检查器会自动识别a的类型为int
b = func({'x': 1, 'y': '2'}, 'y')  # b的类型会被正确推断为str

关键细节解释： #

1. 类型变量TKey：用TypeVar创建的这个变量被bound=Literal['x', 'y']约束，意味着它只能取MyType定义的键值，类型检查器会直接阻止传入非法键（比如'z'），提前避免错误。
2. 返回类型MyType[TKey]：这是实现动态类型推断的核心，和TypeScript里的写法几乎一致！它告诉类型检查器：返回值的类型就是MyType中对应键k的类型。

优化：避免魔法字符串 #

你之前写的Key类可以简化成字面量常量，既避免拼写错误，又保持类型安全：

X_KEY: Literal['x'] = 'x'
Y_KEY: Literal['y'] = 'y'

a = func({'x': 1, 'y': '2'}, X_KEY)  # 同样能正确推断为int类型

为什么动态修改__annotations__行不通？ #

你提到的动态修改函数注解的方法，静态类型检查器（比如mypy、pyright、PyCharm内置检查）是基于静态代码分析的，不会执行运行时的动态修改逻辑，所以这种方法无法被类型检查器识别，必须用静态的泛型约束方式。

兼容性说明 #

这种写法在Python 3.8及以上版本都能正常工作，主流类型检查器都能正确推断返回值类型。如果用的是Python 3.10+，还可以用更简洁的联合类型写法：bound=Literal['x'] | Literal['y']，效果完全一致。

备注：内容来源于stack exchange，提问作者Tyler Wright