无法连接两个numpy数组

Question

给出两个numpy数组a1和a2：

>>>np.shape(a1)
(4465, 5000)
>>>np.shape(a2)
(4465, )

然而，

>>>np.concatenate((a1, a2), axis=1)
ValueError: all the input arrays must have the same number of dimensions

我也尝试过：

np.concatenate((a1, a2), axis=1),
np.concatenate((a1, a2.T), axis=0),
np.concatenate((a1, a2.T), axis=1)

但也有同样的错误。

你能告诉我我的代码有什么问题吗？谢谢！

Answer 1

正如错误消息所述，a1和a2的维度数量不同（可通过属性ndims访问）。使用a2制作a2 = a2[:, None] 2维。您还可以使用更明确的语法a2 = a2[:, np.newaxis]，但它完全等效。