如何在Python中垂直连接两个数组？

Question

我想使用NumPy包在Python中垂直连接两个数组：

a = array([1,2,3,4])
b = array([5,6,7,8])

我想要这样的东西：

c = array([[1,2,3,4],[5,6,7,8]])

我们如何使用concatenate函数来做到这一点？我检查了这两个函数，但结果是相同的：

c = concatenate((a,b),axis=0)
# or
c = concatenate((a,b),axis=1)

这两个功能都具有此功能：

c = array([1,2,3,4,5,6,7,8])

Answer 1

问题在于两者a和b都是一维数组，因此只有一个轴可以将它们连接在一起。

相反，您可以使用vstack（
v 表示 垂直 ）：

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

另外，row_stack是该vstack函数的别名：

>>> np.row_stack((a,b))
array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

还值得注意的是，相同长度的多个数组可以一次堆叠。例如，np.vstack((a,b,x,y))将有四行。

在幕后，vstack通过确保每个数组至少具有两个维度（使用atleast_2D），然后concatenate在第一个轴（axis=0）上调用以连接这些数组，来进行工作。