类型提示npt.NDArray轴的数量

Question

给定轴的数量，我可以将轴的数量指定为类型hint npt吗?NDArray (fromimport numpy.typing as npt)

。如果我知道它是一个3D数组，我怎么做npt.NDArray[3, np.float64]

Answer 1

在Python 3.9和3.10中，下面的代码为我完成了这项工作:

data = [[1, 2, 3], [4, 5, 6]]
arr: np.ndarray[Tuple[Literal[2], Literal[3]], np.dtype[np.int_]] = np.array(data)

这有点麻烦，但您可以遵循numpy问题#16544，以便将来开发更简单的规范。

特别是,现在你必须声明完整的形状和不能只声明数组的排序。

将来应该可以使用ndarray[Shape[:, :, :], dtype]之类的东西。