Создание нового 2D-массива из среза 3D-массива в python?

Question

У меня есть трехмерный массив [256,256,450], который я хотел бы получить двумерный массив из поперечного сечения вдоль оси z. Новый двумерный массив должен начинаться с z = 0 в верхней части и иметь значения на некотором i-й секции, от x = 0 до x = 255. Тогда следующая строка в новом двумерном массиве должна быть такой же для z = 1 и так далее, пока z = 449. Как это можно сделать?

Answer 1

Использовать NumPy * NDArray class и синтаксис синтаксис.

import numpy as np
my_array = np.zeros([256, 256, 450]) # 256x256x450 array
... # Do whatever you want to do to load data in
x_slice = my_array[0,:,:] # A 256x450 array, the first element in x
y_slice = my_array[:,0,:] # A 256x450 array, the first element in y
y_slice = my_array[:,99,:] # A 256 x 450 array, the 100th element in y

Answer 2

import numpy as np
array_3d = np.ones((256, 256, 450))
y_layer = 24 # arbitrary y layer
array_2d = array_3d[:, y_layer, :]