python - 无法连接两个 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[:, None] 将 a2 设为二维。您还可以使用更明确的语法 a2 = a2[:, np.newaxis]，但它是严格等效的。