Numpy阵列形状问题

Question

我已经初始化了这个空的2d np.array

inputs = np.empty((300, 2), int)

我正试图在它上附加一个2d行

inputs = np.append(inputs, np.array([1,2]), axis=0)

但我得到了

ValueError: all the input arrays must have same number of dimensions

Numpy认为它是一个2行0维物体(转置2d)

np.array([1, 2]).shape

(2，)

我哪里出错了？

Answer 1

若要向(300,2)形状数组添加一行，需要(1,2)形状数组。注意匹配的第二维度。

np.array([[1,2]])工作。np.array([1,2])[None, :]和np.atleast_2d([1,2])也是如此。

我鼓励使用np.concatenate。它迫使你更仔细地思考维度。

你真的想从np.empty开始吗？看看它的价值。它们是随机的，而且可能很大。

@Divakar建议np.row_stack。这让我有点困惑，直到我检查发现它只是np.vstack的另一个名字。该函数在执行np.atleast_2d之前通过np.concatenate传递所有输入。因此，最终相同的解决方案--将(2,)数组转换为(1,2)

Answer 2

Numpy需要双括号来声明数组文字，因此

np.array([1,2])

需要的是

np.array([[1,2]])

Answer 3

如果您打算将其作为最后一行附加到inputs中，则只需使用np.row_stack -

np.row_stack((inputs,np.array([1,2])))

请注意，这个np.array([1,2])是一个1D数组。

您甚至可以为相同的结果传递一个2D行版本-

np.row_stack((inputs,np.array([[1,2]])))