np.ndarray 的类型参数

【问题标题】：type parms of np.ndarraynp.ndarray 的类型参数
【发布时间】：2022-06-30 19:25:43
【问题描述】：

当我输入时

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.float64)

pylance 告诉我 a 的类型是：

(variable) a: ndarray[Unknown, Unknown]

我不知道这两个Unknown的意思。

引用存根文件

class ndarray(_ArrayOrScalarCommon, Generic[_ShapeType, _DType_co]):

我还是不知道。我该如何填写？

【问题讨论】：

【解决方案1】：

那些unknown指的是创建的数组的形状和数据类型。

由于np.array 只是np.ndarray(shape=(1,))，它们具有相同的返回类型：ndarray[_ShapeType@ndarray, _DType_co@ndarray]

因此您的代码应该像这样工作：

import numpy as np

a: np.ndarray[int, np.dtype[np.float64]] = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.float64)

其中int 指大小元组数据类型，np.dtype[np.float64] 指数组数据类型

【讨论】：