python - 输入提示二维 numpy 数组

Question

作为this question的后续我有一个函数会返回一个 2D numpy.array固定列但可变行

import numpy.typing as npt

def example() -> npt.ArrayLike:
    data = np.array([[1,2,3],
                     [4,5,6],
                     ...,
                     [x,y,z]])

我如何特别提示返回的数组将是 3专栏作者 N (变量)行？

Answer 1

似乎不可能对 numpy.ndarray 的形状(或数据类型)进行类型提示。此时(2021 年 3 月 16 日)。然而，最近有一些 numpy 的拉取请求正在朝着这个目标努力。

https://github.com/numpy/numpy/pull/17719

makes the np.ndarray class generic w.r.t. its shape and dtype: np.ndarray[~Shape, ~DType]

但是，该 PR 的一个明确非目标是为 numpy.ndarray 制作运行时可下标的别名。 .根据该 PR，这些更改将在以后的 PR 中出现。

https://github.com/numpy/numpy/issues/16544
问题讨论对形状的输入支持。在撰写本文时，它仍然是开放的。

这可以通过 nptyping package 来实现。 ，但这不是 numpy 的一部分。

from typing import Any
from nptyping import NDArray

# Nx3 array with Any data type.
NDArray[(Any, 3), Any]