将3D数组重塑为2D数组Python

Question

如果我有一个3D数组（[4,3,3]），如下所示：

[[0,1,2]   [[9,10,11 ]    [[18,19,20]   [[27,28,29] 
 [3,4,5]    [12,13,14]     [21,22,23]    [30,31,32]
 [6,7,8]] , [15,16,17]] ,  [24,25,26]] , [33,34,35]]

我如何将它转换成一个二维数组（[6,6]），这样数组的前半部分位于160x160的上半部分，而下半部分位于底部：

^{pr2}$

我的阵列创建：

qDCTReversed = np.zeros((400,8,8), dtype=np.int)

我需要一个（160160）阵列。在

Answer 1

您可以通过循环列表来完成此操作：

a = [[[ 0, 1, 2], [ 9,10,11]],
     [[ 3, 4, 5], [12,13,14]],
     [[ 6, 7, 8], [15,16,17]],
     [[18,19,20], [27,28,29]],
     [[21,22,23], [30,31,32]],
     [[24,25,26], [33,34,35]]]

b = [[i for j in k for i in j ] for k in a]
print(b)

输出：

^{pr2}$

Answer 2

您要求的重塑可以通过以下方式完成：

x = np.arange(36).reshape((4,3,3))
np.vstack(np.hstack(x[2*i:2+2*i]) for i in range(x.shape[0]//2))
>>array([[ 0,  1,  2,  9, 10, 11],
   [ 3,  4,  5, 12, 13, 14],
   [ 6,  7,  8, 15, 16, 17],
   [18, 19, 20, 27, 28, 29],
   [21, 22, 23, 30, 31, 32],
   [24, 25, 26, 33, 34, 35]])

Answer 3

使用无for-循环的快速单线解决方案是：

# initialization
qDCTReversed = np.arange(4*3*3).reshape((4,3,3)) 

# calculation
qDCTReversed = qDCTReversed.reshape((2,2,3,3)).transpose((0,2,1,3)).reshape((6,6))

或者对于(400,8,8)数组：

^{pr2}$

速度比较：

Mstaino's answer:0.393毫秒

yatu's answer:0.138毫秒

这个答案：0.016毫秒