使用pandas函数传axis="rows"时mypy报错的解决方法

当你运行pd.concat(dfs, axis="rows")、pd.median(dfs, axis="rows")这类带axis参数的pandas函数时，mypy抛出error: No overload variant of "concat" matches argument types "list[DataFrame]", "str"错误，本质原因是pandas的类型注解定义里，axis参数仅接受整数（0代表行、1代表列）——虽然运行时pandas支持"rows"这类字符串别名，但mypy不认可该类型匹配。

以下是几种不影响代码功能的解决办法：

• 替换为整数轴参数：直接把axis="rows"改成axis=0，axis="columns"改成axis=1。这是最规范的处理方式，完全符合类型定义，mypy会直接通过，示例：

  pd.concat(dfs, axis=0)
  pd.median(dfs, axis=0)
  

• 添加类型忽略注释：如果不想修改代码逻辑，可在报错行末尾加上# type: ignore[call-overload]，让mypy跳过该重载匹配检查，示例：

  pd.concat(dfs, axis="rows") # type: ignore[call-overload]
  

  注意：这是妥协方案，会屏蔽该行的重载相关类型错误，尽量少用，避免漏掉其他潜在问题。

• 升级pandas版本：部分较新版本的pandas已更新类型注解，支持字符串格式的axis参数。可以尝试升级到最新稳定版，比如执行pip install --upgrade pandas，之后mypy可能就不会再报这个错误了。

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