如何將多個嵌套的numpy數組更改為一個多維數組？

Question

我有一個帶有這些嵌套的numpy數組的numpy a = np.load('test.npy')文件：

In [21]: a.shape
Out[21]: (6886, 3)

In [22]: a[0].shape
Out[22]: (3,)

In [23]: a[0][0].shape
Out[23]: (787, 6)

有沒有簡單的方法可以將a更改a形狀為(6886, 3, 787, 6)的4維數組？

Answer 1

我不願意這樣做，但是想到的只是制作所需形狀的第二個數組並將數據切成該數組。 我不得不承認，我在理解每個子陣列的形狀時遇到了困難……這似乎與直覺相反。 無論如何，此解決方案將很慢，但是您只能執行一次並保存陣列，而不再執行。

import numpy as np

a = np.load('test.npy')
b = np.full((6886, 3, 787, 6), np.nan)

for row in range(6886):
    for col in range(3):
        tmp = a[row][col]
        b[row, col, :tmp.shape[0], :tmp.shape[1]] = tmp

這有意義嗎？