How can I get stub files for `matplotlib`, `numpy`, `scipy`, `pandas`, etc.?

Question

I know that the stub files for built-in Python library for type checking and static analysis come with mypy or PyCharm installation. How can I get stub files for matplotlib , numpy , scipy , pandas , etc.?

Answer 1

Type stubs are sometimes packaged directly with the library. Otherwise there can be some external libraries to provide them.

Numpy

Starting with numpy 1.20 type stubs will be included in numpy. See this changelog and this PR adding them

Before that they could added with the library https://github.com/numpy/numpy-stubs

Pandas and Matplotlib

There is no official support for these libraries stubs but you can find unofficial stubs in this project: https://pypi.org/project/data-science-types/

You can either install this library as a dependency or copy only the relevant part in the type stubs folder of your project.

Answer 2

If you are using VSCode and Pylance extension with strict type checking enabled, you can generate stub files automatically by just pressing quickfix option and press "Create Type Stub for " module " "

Answer 3

As of July 2022 there are official pandas stubs: https://github.com/pandas-dev/pandas-stubs

This is the home for pandas typing stubs supported by the pandas core team. The stubs are likely incomplete in terms of covering the published API of pandas.