Pandas的Mypy/typeshed存根

Question

只是检查一下是否有人正在监听，是否已经为pandas生成了一组类型/类型的存根。我天真地在本地pandas安装上运行stubgen，这产生了一些错误。我可以从一开始就这么做，但我希望其他人能把球推得更远。(在GitHub上没有什么明显的发现，尽管存根有一个旧的标签。)

Answer 1

到目前为止(2021年3月)，a dedicated pandas stub project正在进行中，请参阅。然而，它不是由pandas团队开发的(感谢@remeus指出这一点)。data-science-types项目已弃用。在GitHub they write上

⚠️此项目已基本停止开发⚠️

pandas团队和numpy团队都在将类型存根集成到他们的代码库中，我们看不到与他们竞争的意义。

NumPy存根过去是在一个单独的项目中，但现在have been merged到主git代码库：

这些存根已经合并到NumPy中，所有进一步的开发都将在NumPy主存储库中进行。我们欢迎您的贡献！

Answer 2

我还没有找到pandas的存根。然而，有人为NumPy has created了一些。

一种临时的选择是根据mypy如何看待pandas对象来定义一个自定义类型。当我在reveal_type(df)中包装一个DataFrame对象并运行mypy时，它揭示了可识别的类型签名是

Union[builtins.dict[Union[builtins.str, builtins.int], builtins.dict[Any, Any]], Any]

然而，对于Series对象，它显示了同样的情况，因此它不是很准确。

Answer 3

截至2020年6月，熊猫还没有树桩，但在那个方向上有is some progress。到目前为止，您可以使用data-science-types包(on GitHub)，它为pandas、numpy和matplotlib提供了存根文件，不过这也是一项正在进行的工作。