python - 如何将数组从ABAB更改为AABB

Question

假设我有一个大矩阵，其列采用这种方式[A1,B1,C1,A2,B2,C2,A3,B3,C3]。如何将其重新排列为[A1,A2,A3,B1,B2,B3,C1,C2,C3]的格式。在这种情况下，两个A之间的步长是3，如果步长为n呢？

x=np.array(
[[ 1.1, 2.1 , 3.1,  1.2,  2.2, 3.2],
 [ 4.1, 5.1,  6.1,  4.2,  5.2, 6.2]])

x[:,[0,3,1,4,2,5]] # is there a way to do it not manually?

期望的输出

array([[1.1, 1.2, 2.1, 2.2, 3.1, 3.2],
       [4.1, 4.2, 5.1, 5.2, 6.1, 6.2]])

Answer 1

使用切片和 numpy.hstack :

>>> x = np.array([
...     [ 1.1, 2.1 , 3.1,  1.2,  2.2, 3.2],
...     [ 4.1, 5.1,  6.1,  4.2,  5.2, 6.2],
... ])
>>> n = 3
>>> np.hstack(x[:,i::n] for i in range(n))
array([[ 1.1,  1.2,  2.1,  2.2,  3.1,  3.2],
       [ 4.1,  4.2,  5.1,  5.2,  6.1,  6.2]])