Python作为动态类型语言，变量类型在运行时才会确定，mypy作为静态类型检查工具，核心能力之一就是在不执行代码的前提下，通过类型推断确定变量、函数返回值等的类型，从而提前发现类型不匹配的问题。
mypy类型推断的基础前提
mypy的类型推断不是凭空进行的，它首先依赖明确的类型信息作为锚点，这些锚点包括：
- 显式类型标注：比如变量声明时的
name: str = "test"，函数参数和返回值的类型标注 - 内置类型与标准库类型定义：mypy内置了Python基础类型、标准库的类型存根文件，知道
len()函数返回int，list.append()方法接收与列表元素类型一致的参数 - 类型推断的初始上下文：比如函数调用时传入的参数类型，会作为推断函数返回值类型的上下文依据
核心推断流程拆解
1. 单变量类型推断
当变量没有显式类型标注时，mypy会根据赋值语句的右侧表达式推导类型。比如下面的代码：
# 右侧是字符串字面量，推断name为str类型
name = "张三"
# 右侧是列表字面量，元素都是int，推断scores为list[int]类型
scores = [90, 85, 92]
# 右侧是函数调用，根据函数标注的返回值推断result类型
def add(a: int, b: int) -> int:
return a + b
result = add(1, 2) # result被推断为int类型
2. 函数返回值类型推断
如果函数没有显式标注返回值类型，mypy会分析函数体内所有可能的返回路径，取所有返回值的类型的并集作为返回值类型。如果返回值类型不一致且没有共同父类，会标记为object类型或者报错。示例如下：
def get_value(flag: bool):
if flag:
return 10 # 返回int
else:
return "hello" # 返回str
# mypy推断返回值类型为int | str
3. 上下文类型推断
上下文类型推断是指mypy会根据表达式所在的上下文环境，反向推导表达式的类型。最典型的场景是赋值给容器类型，或者作为函数参数传入时：
from typing import List
# 上下文要求scores是List[int]，因此空列表被推断为List[int]，后续只能添加int元素
scores: List[int] = []
scores.append(90) # 合法
scores.append("语文") # mypy会报错，因为上下文要求元素是int
def print_num(num: int) -> None:
print(num)
# 传入的lambda需要匹配参数类型为int的要求，因此x被推断为int
print_num(lambda x: x + 1)
4. 泛型类型推断
对于泛型函数或泛型类，mypy会根据传入的实际参数类型推断泛型的具体类型参数。比如下面的泛型函数示例：
from typing import TypeVar, List
T = TypeVar('T')
def first_element(arr: List[T]) -> T:
return arr[0]
# 传入List[int]，推断T为int，返回值类型为int
num = first_element([1, 2, 3])
# 传入List[str]，推断T为str，返回值类型为str
word = first_element(["a", "b", "c"])
类型推断的局限性
mypy的类型推断并不是万能的，在一些复杂场景下可能无法准确推断类型，需要开发者手动标注：
- 动态创建的对象、使用
exec或eval执行的代码，mypy无法获取类型信息 - 复杂的继承关系和多重分派场景，推断结果可能不符合预期
- 涉及
Any类型的操作，mypy会跳过类型检查，推断结果也会是Any
遇到推断不准确的情况，开发者可以添加显式类型标注，或者调整代码结构让推断逻辑更清晰，从而让mypy发挥更好的类型检查效果。