将此（4,4）numpy ndarray重塑为（2,2,4）ndarray

Question

我有这个形状为nd的numpy ndarray (4,4)。

[
    [1.07 1.16 1.00 1.11]
    [1.13 1.19 1.11 1.17]
    [1.17 1.17 1.13 1.16]
    [1.14 1.16 1.03 1.04]
]

我想将其重塑成一个(2,2,4)的ndarray，它应该看起来像这样。

[
    [
        [1.07 1.16 1.00 1.11]
        [1.13 1.19 1.11 1.17]
    ]
    [
        [1.17 1.17 1.13 1.16]
        [1.14 1.16 1.03 1.04]
    ]
]

我正在使用python v3.6

Answer 1

您可以使用reshape：

import numpy as np
nd = np.array([
    [1.07, 1.16, 1.00, 1.11],
    [1.13, 1.19, 1.11, 1.17],
    [1.17, 1.17, 1.13, 1.16],
    [1.14, 1.16, 1.03, 1.04]])

nd_new = np.reshape(nd, (2,2,4))