NumPy Array Conversion

Question

Я новичок в python и пытаюсь быстро (m, n, 1) преобразовать многомерный массив в (m, n), как мне это сделать?

Также учитывая массив (m, n, k), как я могу разделить его на массивы k (m, n)?(каждый из k членов принадлежит отдельному массиву)

Answer 1

Для изменения формы массива a вы можете использовать a.reshape(m,n).

Чтобы разделить массив a по измерению глубины, вы можете использовать numpy.dsplit(a, a.shape[2]).

https://docs.scipy.org/doc/numpy/reference/generated/numpy.split.html https://docs.scipy.org/doc/numpy/reference/generated/numpy.dsplit.html#numpy.dsplit

Answer 2

Для преобразования массива NumPy arr с формой (m, n, 1) в форму (m, n) просто используйте:

arr = arr.reshape(m, n)

Вы можете получить список (m, n) -образные массивы из массива (m, n, k) arr_k по:

array_list = [arr_k[:, :, i] for i in range(arr_k.shape[2])]