如何结合两个不同维度的numpy数组？

Question

我有两个numpy数组如下：

a=np.array([[1,2],[3,4]])
b=np.array([[3,4],[1,2],[3,4]])

我想要一个结果：

np.array([[[1,2],[3,4]],[[3,4],[1,2],[3,4]]])

我尝试了np.concatenate和np.stack，但第一个是不可行的，第二个非常慢。 如何更快地解决这个问题？

Answer 1

尝试：

import numpy as np 
r = np.array([np.array([[1,2],[3,4]]), np.array([[5,6],[7,8]])])
print(r)

正如@COLDSPEED指出的那样，结果将不会完全符合您的预期，因为从数据集的角度来看，存储具有不同维度的行是没有意义的。这就是结果类似于数组列表的原因。这可能表明您尝试对工作流程采取的方法并不完全合适。也许你可以尝试重新思考你的数据模型。