python - Numpy 3d 数组按最外层索引转换为 2d 数组

Question

我有一个二维数组的数组，例如

+------+    +------+
|      |    |      |
|  A   |    |  B   |
|      |    |      |
+------+    +------+

我想“删除”最外面的括号，如获取

+------+------+
|      |      |
|  A   |  B   |
|      |      |
+------+------+

例如我有

[[[1,1,1],[2,2,2]],[[3,3,3],[4,4,4]]]

我想要得到

[[1,1,1,3,3,3],[2,2,2,4,4,4]]

换句话说，我需要通过将这 7 附加到链中来将形状 (7,3,1000) 变成 (3,7000) 的数组

如何去做？

Answer 1

一种方法是在第一个和第二个轴之间交换轴，然后 reshape 以合并最后两个轴 -

arr.swapaxes(0,1).reshape(arr.shape[1],-1)

示例运行 -

In [9]: arr = np.array([[[1,1,1],[2,2,2]],[[3,3,3],[4,4,4]]])

In [10]: arr.swapaxes(0,1).reshape(arr.shape[1],-1)
Out[10]: 
array([[1, 1, 1, 3, 3, 3],
       [2, 2, 2, 4, 4, 4]])