如何重新排列 Numpy 数组：2D 到 3D？

Question

我有以下数组。

>>> import numpy as np

>>> a = np.arange(100).reshape(20,5)

>>> a

array([[ 0,  1,  2,  3,  4],

       [ 5,  6,  7,  8,  9],

       [10, 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, 59],

       [60, 61, 62, 63, 64],

       [65, 66, 67, 68, 69],

       [70, 71, 72, 73, 74],

       [75, 76, 77, 78, 79],

       [80, 81, 82, 83, 84],

       [85, 86, 87, 88, 89],

       [90, 91, 92, 93, 94],

       [95, 96, 97, 98, 99]])

我如何将其重塑为：

>>> a

array([[[ 0,  1,  2,  3,  4],

        [ 5,  6,  7,  8,  9],

        [10, 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, 59],

        [60, 61, 62, 63, 64],

        [65, 66, 67, 68, 69],

        [70, 71, 72, 73, 74]],

       [[75, 76, 77, 78, 79],

        [80, 81, 82, 83, 84],

        [85, 86, 87, 88, 89],

        [90, 91, 92, 93, 94],

        [95, 96, 97, 98, 99]]])

Answer 1

你只需要重塑：a = a.reshape(4,5,5)