如何为numpy向量的每个元素分配大小不同的3d numpy数组？

Question

这里是情况：

x.shape = (20,)

x1 = x [0,]
x2 = x [1,]
x3 = x [2,]

...

x1，x2，x3 ...的形状都不同。

例如，

x1.shape = (300, 400, 3)
x2.shape = (280, 520, 3)
x3.shape = (330, 400, 3)

我有一个不同大小的3d numpy数组

我想将其分配给向量的每个元素x。

我该怎么办？

Answer 1

首先制作您的元素，然后根据它们创建x数组

x = numpy.array([x1, x2, x3])

现代的numpy将自动使用dtype=object而在以前的版本中，您必须显式添加dtype=object 。

不过，这与使用普通的Python list并没有太大区别：

x = [x1, x2, x3]