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

Question

如果我有这样的 ([4,3,3]) 3D 数组：[[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 的上半部分，而第二半位于底部：[[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]]我的阵列创建：qDCTReversed = np.zeros((400,8,8), dtype=np.int)我需要一个 (160,160) 数组。

Answer 1

使用无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)数组：

qDCTReversed.reshape((20,20,8,8)).transpose((0,2,1,3)).reshape((160,160))

速度对比：

Mstaino 的回答：0.393 毫秒

yatu 的回答：0.138 毫秒

这个答案：0.016 毫秒

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

您可以通过循环遍历列表来做到这一点：

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)

输出：

[ 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]