mypy校验numpy时索引NDArray返回Any类型，能否返回对应数组类型？

可以实现索引操作返回值和原数组保持一致的类型，你遇到的mypy推导为Any的问题是缺少numpy官方mypy插件导致的，可通过以下两种方式解决：

• 方法1：启用numpy自带的mypy类型插件
  首先确保使用的numpy版本≥1.21，之后在mypy配置文件中添加插件配置即可。
  以pyproject.toml为例，添加如下配置：

  [tool.mypy]
  plugins = ["numpy.typing.mypy_plugin"]
  

  配置完成后重新运行mypy，你示例中的sub_b就会被正确推导为NDArray[np.int64]类型，和原数组b的类型完全一致。

• 方法2：手动标注返回值类型
  如果你不想全局启用numpy的mypy插件，也可以直接给索引操作后的变量手动指定类型：

  sub_b: NDArray[np.int64] = b[:1]
  

  这种方式也能让mypy识别到正确的类型，不会返回Any。

目前numpy的mypy插件已经支持常规切片、布尔索引、整数数组索引等常见索引场景的类型推导，只有部分非常复杂的自定义花式索引场景可能仍需要手动标注类型。

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