NumPy數組轉換

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/produced/numpy.split.html https://docs.scipy.org/doc/numpy/reference/genic/numpy.dsplit.html#numpy.dsplit

Answer 2

要將形狀為（m，n，1）的NumPy Array arr重塑為形狀（m，n），只需使用：

arr = arr.reshape(m, n)

您可以通過以下方式從（m，n，k）形數組arr_k中獲得（m，n）形數組的arr_k ：

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