Mypy无法检测列表推导式中元组元素类型错误，如何正确校验？

问题背景 #

在Python 3.11 + Mypy 1.6.0环境下，以下代码存在明显的类型不匹配错误，但Mypy未能识别：

MyTupleType = tuple[int, str, bool]
Tuples = list[MyTupleType]
 
def findAllByInt(values : Tuples, ref : int) -> Tuples:
  return [v for v in values if v[1] == ref]  # 错误点：用str类型的v[1]和int类型的ref做比较
 
vals : Tuples = [ (1, 'a', True), (2, 'b', False),
                  (3, 'c', True), (1, 'd', False) ]
 
print( findAllByInt(vals, 1) )

错误原因是代码误用元组索引1对应的str类型元素，与int类型的ref进行比较，正确索引应为0（对应int类型元素）。

目前已知的解决方案是编写元组元素提取函数，此时Mypy能正确检测到类型不匹配：

MyTupleType = tuple[int, str, bool]
Tuples = list[MyTupleType]
 
def intOfMyTuple(t : MyTupleType) -> int:
  return t[0]
 
def findAllByInt(values : Tuples, ref : int) -> Tuples:
  return [v for v in values if intOfMyTuple(v) == ref]
 
vals : Tuples = [ (1, 'a', True), (2, 'b', False),
                  (3, 'c', True), (1, 'd', False) ]
 
print( findAllByInt(vals, 1) )

无需编写提取函数的替代方案 #

不需要为每个元组及索引都编写提取函数，有两种方式可以让Mypy直接在第一份代码中检测到类型错误：

1. 使用typing.NamedTuple替代普通元组 #

将普通元组替换为具名元组，通过属性名而非索引访问元素，Mypy会直接校验类型匹配性：

from typing import NamedTuple

class MyTupleType(NamedTuple):
    id: int
    name: str
    flag: bool

Tuples = list[MyTupleType]
 
def findAllByInt(values : Tuples, ref : int) -> Tuples:
  return [v for v in values if v.name == ref]  # Mypy会报错：str类型与int类型无法直接比较
 
vals : Tuples = [ MyTupleType(1, 'a', True), MyTupleType(2, 'b', False),
                  MyTupleType(3, 'c', True), MyTupleType(1, 'd', False) ]
 
print( findAllByInt(vals, 1) )

这种方式不仅能触发Mypy的类型检查，还能提升代码可读性，避免索引混淆的问题。

2. 启用Mypy的strict-equality选项 #

在Mypy配置文件（如pyproject.toml或mypy.ini）中开启strict-equality选项，或直接启用strict模式，Mypy会对不同类型的相等比较抛出错误：

以pyproject.toml为例，添加以下配置：

[tool.mypy]
strict-equality = true

此时运行Mypy检查第一份代码，会直接提示Invalid comparison between "str" and "int"的错误，无需修改原有代码结构。

总结 #

编写提取函数是可行方案，但并非必须。使用具名元组既能优化代码可读性，又能让类型检查更直观；调整Mypy配置则可以在不改动代码结构的前提下，直接检测到类型不匹配问题。

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