如何將 numpy 1D 數組插入 numpy 3D 數組？

Question

所以我有一個 3 維 numpy 數組，我想在其中插入一個 1 維 numpy 數組。 我該怎么做？
例如，這是我的 3D 數組，我想插入 [2,2,2]

[[[1,1,1],
  [3,3,3],
  [4,4,4]],
 [[5,5,5],
  [6,6,6],
  [7,7,7]]]

所以它看起來像這樣：

[[[1,1,1],
  [2,2,2],
  [3,3,3],
  [4,4,4]],
 [[5,5,5],
  [6,6,6],
  [7,7,7]]]

我該怎么做？

Answer 1

您不能使用標准 numpy 數組執行此操作，因為它們必須保持為矩形。 潛在地， 您可以使用dtype=object創建一個，但在我看來，這會失去numpy的效率。

也許你最好使用常規列表？

l = [[[1,1,1],
      [3,3,3],
      [4,4,4]],
     [[5,5,5],
      [6,6,6],
      [7,7,7]]]
l[0].insert(1, [2,2,2])

將l修改為：

l = [[[1,1,1],
      [2,2,2],
      [3,3,3],
      [4,4,4]],
     [[5,5,5],
      [6,6,6],
      [7,7,7]]]