python - 垂直连接两个 NumPy 数组

Question

我尝试了以下方法:

>>> a = np.array([1,2,3])
>>> b = np.array([4,5,6])
>>> np.concatenate((a,b), axis=0)
array([1, 2, 3, 4, 5, 6])
>>> np.concatenate((a,b), axis=1)
array([1, 2, 3, 4, 5, 6])

但是，我希望至少有一个结果看起来像这样

array([[1, 2, 3],
       [4, 5, 6]])

为什么不垂直拼接？

Answer 1

因为a和b都只有一个轴，因为它们的形状是(3)，而轴参数具体指的是要连接的元素的轴。

这个例子应该阐明 concatenate 对轴做了什么。取两个带有两个轴的向量，形状为 (2,3):

a = np.array([[1,5,9], [2,6,10]])
b = np.array([[3,7,11], [4,8,12]])

沿第一个轴连接(第一个轴的行，然后是第二个轴的行):

np.concatenate((a,b), axis=0)
array([[ 1,  5,  9],
       [ 2,  6, 10],
       [ 3,  7, 11],
       [ 4,  8, 12]])

沿第 2 轴连接(第 1 列，然后第 2 列):

np.concatenate((a, b), axis=1)
array([[ 1,  5,  9,  3,  7, 11],
       [ 2,  6, 10,  4,  8, 12]])

要获得你呈现的输出，你可以使用 vstack

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

你仍然可以使用 concatenate 来做到这一点，但你需要先 reshape 它们:

np.concatenate((a.reshape(1,3), b.reshape(1,3)))
array([[1, 2, 3],
       [4, 5, 6]])

最后，正如评论中所建议的， reshape 它们的一种方法是使用 newaxis:

np.concatenate((a[np.newaxis,:], b[np.newaxis,:]))