numpy按列连接两个维数不同的数组

合并两个维数不同的ndarray

假设我们有一个3×2 numpy数组：

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

现在需要把它与一个一维数组：

y = array(([7, 8,9]))

通过将其添加到行的末尾，连接为一个3×3 numpy数组，如下所示：

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

在numpy中按列连接的方法是：

hstack((x,y))

但是这不行，会报错：

ValueError: arrays must have same number of dimensions

解决方法有两种：

方法一：

>>> x = np.array([[1,2],[3,4],[5,6]])
>>> y = np.array([7,8,9])
>>> np.hstack((x, np.array(([y])).T ))
array([[1, 2, 7],
       [3, 4, 8],
       [5, 6, 9]])

方法二：

>>> x = np.array([[1,2],[3,4],[5,6]])
>>> y = np.array([7,8,9])
>>> np.column_stack((x,y))
array([[1, 2, 7],
       [3, 4, 8],
       [5, 6, 9]])