Как извлечь вектор из массива 3d numpy?

У меня есть набор numpy.arrays из NXM (два измерения: Range и Azimuth). Мне нужно сформировать стек из трех измерений и извлечь один размерный вектор для вычисления ковариационной матрицы (красные векторы на картинке).

введите описание изображения здесь

Как я делаю это эффективно и легко в Python?

Вы можете сделать массив 3D numpy довольно легко, а затем просто используйте индексирование, чтобы вытащить нужные вам биты:

stackOfImages = np.array((image1, image2)) #iterate over these if many more redData = stackOfImages[:, N-1, M-1]