从展平的Numpy重塑

2admgd59 于 2023-04-30 发布在其他

关注(0)|答案(1)|浏览(129)

一些代码会导致numpy数组：array([[2, 2, 1, 0, 4, 2, 3, 3]])
我想从中恢复：

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

         [[2, 0], 
          [2, 3]] ])

即：将跳过的元素放入两个矩阵中。
请告知如何实现任意数量的最终矩阵（在玩具例子中为2）和它们的维数（总是彼此相等;玩具示例中的2x2）。
我尝试了reshape(2,2,2,1)，结果失败了：

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

        [[1],
         [0]]],

       [[[4],
         [2]],

        [[3],
         [3]]]])

numpy

来源：https://stackoverflow.com/questions/76059646/numpy-reshaping-from-flattened

1条答案

按热度按时间

icnyk63a1#

In [195]: arr = np.array([[2, 2, 1, 0, 4, 2, 3, 3]])

In [196]: arr.reshape(2,2,2)
Out[196]: 
array([[[2, 2],
        [1, 0]],

       [[4, 2],
        [3, 3]]])

第一次猜测转置-

In [197]: arr.reshape(2,2,2).transpose(2,1,0)
Out[197]: 
array([[[2, 4],
        [1, 3]],

       [[2, 2],
        [0, 3]]])

不完全是，切换最后2个维度可以纠正：

In [198]: arr.reshape(2,2,2).transpose(2,0,1)
Out[198]: 
array([[[2, 1],
        [4, 3]],

       [[2, 0],
        [2, 3]]])

赞(0）回复(0）举报 2023-04-30

我来回答

从展平的Numpy重塑

1条答案

相关问题

热门标签

最新问答