我想创建一个3D NumPy数组,序列号如下:
[[[11 27 43]
[12 28 44]
[13 29 45]
[14 30 46]]
[[15 31 47]
[16 32 48]
[17 33 49]
[18 34 50]]
[[19 35 51]
[20 36 52]
[21 37 53]
[22 38 54]]
[[23 39 55]
[24 40 56]
[25 41 57]
[26 42 58]]]
我做了:A = np.arange(11, 59).reshape((4, 4, 3))
但是我得到了这个:
[[[11 12 13]
[14 15 16]
[17 18 19]
[20 21 22]]
[[23 24 25]
[26 27 28]
[29 30 31]
[32 33 34]]
[[35 36 37]
[38 39 40]
[41 42 43]
[44 45 46]]
[[47 48 49]
[50 51 52]
[53 54 55]
[56 57 58]]]
所以这不是我想要的顺序。我已经做了一些额外的步骤,以获得正确的三维阵列。首先,我将这些数字形成一个二维数组:A = np.arange(11, 59).reshape((-1, 4))
来得到:
[[11 12 13 14]
[15 16 17 18]
[19 20 21 22]
[23 24 25 26]
[27 28 29 30]
[31 32 33 34]
[35 36 37 38]
[39 40 41 42]
[43 44 45 46]
[47 48 49 50]
[51 52 53 54]
[55 56 57 58]]
然后,我将2D数组拆分并堆叠起来,得到我想要的3D数组:A = np.dstack(np.vsplit(A, 3))
[[[11 27 43]
[12 28 44]
[13 29 45]
[14 30 46]]
[[15 31 47]
[16 32 48]
[17 33 49]
[18 34 50]]
[[19 35 51]
[20 36 52]
[21 37 53]
[22 38 54]]
[[23 39 55]
[24 40 56]
[25 41 57]
[26 42 58]]]
现在我想知道是否有一个更优雅和直接的方法来达到同样的结果。谢谢你。你知道吗
获取远程阵列,重塑并排列轴-
另一种排列轴的方法是使用
np.moveaxis
-^{}
相关问题 更多 >
编程相关推荐