Concat两个不同维度的阵列numpy

Question

我正在尝试连接两个numpy数组以添加额外的列：array_1为(569, 30)而array_2为(569, )

combined = np.concatenate((array_1, array_2), axis=1)

我认为如果我设置axis=2以便它将垂直连接，这将有效。结尾应该是569 x 31阵列。

我得到的错误是ValueError: all the input arrays must have same number of dimensions

有人可以帮忙吗？

THX！

Answer 1

您可以使用numpy.column_stack：

np.column_stack((array_1, array_2))

将1-d数组隐式转换为2-d，因此与@umutto所评论的np.concatenate((array_1, array_2[:,None]), axis=1)相当。

a = np.arange(6).reshape(2,3)
b = np.arange(2)

a
#array([[0, 1, 2],
#       [3, 4, 5]])

b
#array([0, 1])

np.column_stack((a, b))
#array([[0, 1, 2, 0],
#       [3, 4, 5, 1]])

Answer 2

您可以简单地使用numpy的{​​{1}}函数。

例如

hstack