二维数组的 Numpy 特定重塑顺序

Question

我有一个 4x3 矩阵，我需要将其重塑为具有特定顺序的 2x6 矩阵：

初始： |1 5 9| |2 6 10| |3 7 11| |4 8 12|

想要： |1 2 5 6 9 10| |3 4 7 8 11 12|

我尝试使用 numpy，但顺序不是我所期望的：

import numpy as np
a=np.array([[1,5,9],[2,6,10],[3,17,11],[4,8,12]])
b=a.reshape(2,6)
print(a)
print(b)

Answer 1

Reshape, permute and reshape-

In [51]: n = 2 # no. of rows to select from input to form each row of output

In [52]: a.reshape(-1,n,a.shape[1]).swapaxes(1,2).reshape(-1,n*a.shape[1])
Out[52]: 
array([[ 1,  2,  5,  6,  9, 10],
       [ 3,  4, 17,  8, 11, 12]])

一种解释：

1. 沿第一个轴剪切以得到3D 数组，这样我们就可以沿第二个轴选择n。将第二个轴与最后一个（第三个）轴交换，以便我们将第二个轴推回到最后。

2. 重塑为 2D 以合并最后两个轴。这就是我们的输出！

有关更深入的信息，请参阅链接的问答。

---

如果给定输出中的行数 -

In [54]: nrows = 2 # number of rows in output

In [55]: a.reshape(nrows,-1,a.shape[1]).swapaxes(1,2).reshape(nrows,-1)
Out[55]: 
array([[ 1,  2,  5,  6,  9, 10],
       [ 3,  4, 17,  8, 11, 12]])