调整二维 numpy 数组的大小，同时将子列表视为单独的值

Question

[[-2. -2.]
 [ 2. -2.]
 [-2.  2.]
 [ 2.  2.]]

我有一个像这样的 np 数组，我希望它的形状像

[[-2. -2.][ 2. -2.]
 [-2.  2.][ 2.  2.]]

即作为大小为 2x2 的网格，其中每个元素都是一个“有序对”。

Answer 1

您可能需要“转置”或“重新排序”或“排列”轴（所有相似词表示同一事物）。您可以使用

np.ndarray.tranpose

或

np.ndarray.swapaxes

示例：

import numpy as np

arr = np.array([
    [-2, -2],
    [ 2, -2],
    [-2,  2],
    [ 2,  2]
])

out = arr.reshape(2, 2, 2).swapaxes(0, 1)

出：

array([[[-2, -2],
        [-2,  2]],

       [[ 2, -2],
        [ 2,  2]]])