Numpy从扁平化高级重塑

gr8qqesn 于 2023-04-21 发布在其他

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

一些代码会导致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-advanced-reshaping-from-flattened

1条答案

按热度按时间

uqdfh47h1#

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-21

我来回答

Numpy从扁平化高级重塑

1条答案

相关问题

热门标签

最新问答