Numpy矩陣廣播問題

Question

data是一個 1256x277 矩陣

np.array_split(ary=data, indices_or_sections=10, axis=1)給我們一個 1256x28 數據的列表

但是當我這樣做時：

sub_data_ndarray = np.array(object=np.array_split(ary=data, indices_or_sections=10, axis=1))

我得到錯誤：

ValueError：無法將輸入數組從形狀（1256,28）廣播到形狀（1256）

?

Answer 1

嘗試使用np.stack ：

A = [1,2,3,4,5,6,7,8,9,0]
B = [1,2,3,4,5,6,7,8,9,0]
C = np.stack((A, B))