Как добавить измерение в массив numpy и скопировать измерение из другого массива numpy

Question

У меня есть массив numpy с формой (128, 8). Я хочу добавить дополнительное измерение, чтобы оно имело форму (128, 168, 8), и добавить содержимое измерения 168 из другого массива, который имеет форма (128, 168, 8). Я всегда могу изменить положение измерений, если смогу как-то добавить его. Возможно ли это как-то? Я видел методы добавления и конкатенации, но безуспешно.

Answer 1

вы также можете сделать:

small[:,None,:]+big

Добавление None к индексации создает новое измерение размера 1, а добавление в другой больший массив будет передавать измерение small = size = 1 в большие массивы, соответствующие измерению размер (здесь будет 168)

Answer 2

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

Это правильное решение, спасибо!