Python:numpy 形状混淆

Question

我有一个 numpy 数组:

>>> type(myArray1)
Out[14]: numpy.ndarray

>>> myArray1.shape
Out[13]: (500,)

我有另一个数组:

>>> type(myArray2)
Out[14]: numpy.ndarray

>>> myArray2.shape
Out[13]: (500,1)

( 1 ) (500,) 和 (500,1) 有什么区别？

( 2 ) 如何将 (500,) 更改为 (500,1)

Answer 1

(1)(500,)和(500,1)的区别在于第一个是一维数组的形状，而第二个是二维数组的形状，其第2维有长度1. 起初这可能会让人感到困惑，因为其他语言没有这种区别。

(2) 您可以使用 np.reshape 来做到这一点:myArray1.reshape(-1,1)。您还可以使用 np.expand_dims 为数组添加维度:np.expand_dims(myArray1, axis = 1)。