Добавление массива в измерение массива

Question

У меня есть 2 numpy массива img и mask, которые я хочу объединить в один массив.Формы массивов следующие: image.shape = (512, 366, 3) и mask.shape = (512, 366).Я хочу, чтобы конечный массив имел форму (512, 366, 4), чтобы массив mask занимал 4-е измерение.

Какой лучший способ добиться этого, пожалуйста?

Answer 1

Предложение от работ Жюльена:

from numpy import stack
new_image = dstack(img, mask)