如何将一维阵列重塑为具有不同大小的二维数组的三维阵列?

Question

我想重塑这个数组：np.array(np.arange(15))到由 3x3 数组和 3x2 数组构建的 3D 数组。

我尝试使用重塑方法执行此操作，但没有奏效。

我想也许reshape可以得到一些元组。a=np.array(np.arange(15)).reshape(1,((3,2),(3,3)))但后来我看到了它不能。

那我该如何重塑它呢？有没有好办法？

Answer 1

多维数组不能具有不同大小的维度。

但是如果你想要一个元组，你需要将数组分成两部分，第一部分的大小与 3x3 数组匹配，第二部分与 3x2 匹配，此时你将有 2 个一维数组，然后重塑它们

arr1 = arr1.reshape((3,3)) 
arr2 = arr2.reshape((3,2))
tuple = arr1, arr2