Pyright/MyPy 可以推导出 ndarray 条目的类型吗？

Question

如何注释

ndarray

以便 Pyright/Mypy Intellisense 可以推断条目的类型？

???

中

我可以填写什么

x: ??? = np.array([1, 2, 3], dtype=int)

这样

y = x[0]

被识别为整数而不是

Any

？

Answer 1

import numpy.typing as npt

x: npt.NDArray[np.int64] = np.array([1, 2, 3], dtype=int)