如何向 numpy 數組添加維度並從另一個 numpy 數組復制維度

Question

我有一個形狀為 (128, 8) 的 numpy 數組我想添加一個額外的維度，使其具有形狀 (128, 168, 8) 並從另一個形狀為 (128, 168, 8)。 如果我能以某種方式添加它，我總是可以置換尺寸的位置。 這有可能嗎？ 我見過 append 和連接方法，但沒有運氣。

Answer 1

np.expand_dims(smaller_array, axis=1) + bigger_array

是正確的解決方案，謝謝！

Answer 2

你也可以這樣做：

small[:,None,:]+big

將None添加到索引會創建一個大小為 1 的新維度，並且添加到另一個更大的數組會將小的 size=1 維度廣播到更大的 arrays 相應的維度大小（這里將是 168）