python - 如何将 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 数组执行此操作，因为它们必须保持矩形。可能，you could create one with 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]]]