如何向 numpy 数组添加维度并从另一个 numpy 数组复制维度

Question

I have a numpy array with the shape (128, 8) I want to add an extra dimension so it has the shape (128, 168, 8) And add the content of a 168 dimension from another array that has the shape (128, 168, 8). I can always permute the positions of the dimensions if I can somehow add it. Is this possible somehow? I have seen the append and concatenation methods but to no luck.

Answer 1

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

Is the correct solution, thanks!

Answer 2

you can also do:

small[:,None,:]+big

Adding None to indexing creates a new dimension of size 1, and adding to another bigger array will broadcast the small's size=1 dimension to bigger arrays corresponding dimension size (here will be 168)