Numpy соединяется в новом измерении - PullRequest
Новая
       3

Numpy соединяется в новом измерении

0 голосов
/ 04 июля 2019

Я пытаюсь сделать то, что делает этот человек numpy: расширение массивов вдоль новой оси? , но я не хочу повторять тот же массив в новом измерении.Я создаю новый 2D-массив и хочу добавить его к третьему измерению

Я пытался использовать np.stack ((a, b), axis = 2), но массивы должны быть одинаковымиформа.Таким образом, после того, как он сложит первые два массива, фигуры на второй итерации будут (256, 256, 2) и (256, 256), и я получу ValueError: все входные массивы должны иметь одинаковую форму 

a = something #a has shape (256, 256)
for i in np.arange(0,10):
    #calculate b and it also has shape (256,256)
    a = np.stack((a,b), axis=2)
print(a.shape) #should give (256, 256, 10)
...