Python中使用mypy进行类型检查是否为最佳实践？附代码场景疑问

首先明确：用mypy进行静态类型检查绝对属于良好实践。静态类型检查能在代码运行前捕获大量类型相关的bug，提升代码的可读性、可维护性，尤其在大型项目或团队协作场景下，能有效降低沟通成本和潜在的逻辑错误。

解决你的代码报错问题 #

你的代码触发mypy报错，核心原因是类型窄化失效：用type(bar) == float这种判断方式，mypy无法正确识别后续代码中bar的类型已经被窄化为list。替换成isinstance即可解决这个问题——isinstance不仅符合Python的鸭子类型设计（支持子类判断），也是mypy认可的类型窄化方式，能让类型检查器正确推断变量的实际类型。

优化后的基础版本：

from typing import Union

def baz(x: float) -> float:
    return x * 2

def foo(bar: Union[float, list[float]]):
    if isinstance(bar, float):
        bar = [bar]
    # mypy此时能正确识别bar是list[float]，不会报错
    return [baz(i) for i in bar]

关于运行时类型检查的必要性 #

是否需要添加冗余的运行时类型检查，取决于函数的使用场景：

• 公共API/对外接口：建议添加运行时检查。因为静态类型提示对非Python类型检查器的调用者无效，比如有人直接用脚本调用你的函数时可能传入不符合要求的参数，运行时检查能及时抛出明确的错误，避免后续逻辑崩溃。
• 内部函数：如果项目已经开启mypy的严格模式（--strict），且团队成员都遵守类型规范，那么可以省略冗余的运行时检查，减少代码复杂度，保持代码简洁。

优雅的运行时检查写法 #

如果需要添加运行时检查，要避免冗余的pass语句，可以通过合并条件分支优化结构：

from typing import Union

def baz(x: float) -> float:
    return x * 2

def foo(bar: Union[float, list[float]]):
    if isinstance(bar, float):
        bar = [bar]
    elif not isinstance(bar, list):
        raise TypeError(f"参数类型错误：预期float或list[float]，实际传入{type(bar).__name__}")
    
    # 可选：如果需要确保list内的元素都是float，可以添加额外检查
    for item in bar:
        if not isinstance(item, float):
            raise TypeError(f"列表包含非法元素：{item}，预期float类型")
    
    return [baz(i) for i in bar]

总结最佳实践 #

• 优先使用isinstance进行类型判断，而非type()，确保mypy能正确进行类型窄化，解决类型报错问题。
• 坚持使用mypy等静态类型检查工具，建议开启严格模式最大化收益。
• 根据函数的使用场景决定是否添加运行时类型检查：公共接口加，内部函数在静态检查覆盖的前提下可以不加。
• 优化代码结构，避免冗余的分支和pass语句，让代码逻辑更清晰。

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