当我打字时
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.float64)
pylance 告诉我 a 的类型是:
(variable) a: ndarray[Unknown, Unknown]
我不知道这两个未知的意思。
stub 文件的引用
class ndarray(_ArrayOrScalarCommon, Generic[_ShapeType, _DType_co]):
我还是不知道。我该如何填写它们？
最佳答案
那些未知指的是创建的数组的形状和数据类型。
由于 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] 指的是数组数据的类型
关于python - np.ndarray 的类型参数，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/70229215/