轉換ndarray形狀

-1

我正在使用python/Numpy將小圖像存儲到ndarray中。當我試圖將一個ndarray從32,32,1形狀轉換爲1,32,32,1時，我被卡住了。任何幫助？謝謝轉換ndarray形狀

2016-12-30 Mehditation

嘿，是你滿意的答案嗎？ – martianwars

您需要展開numpy數組的維度。使用np.expand_dims。

arr = np.expand_dims(arr, axis=0)

來源

2016-12-30 16:05:37 martianwars

arr[np.newaxis, :, :, :]將工作

來源

2016-12-30 16:05:39 Erotemic

而不是明確地加入你也可以明確地重塑它的軸添加軸：

>>> import numpy as np 
>>> arr = np.ones((32, 32, 1)) # just ones for demonstration purposes 

>>> reshaped = arr.reshape(1, *arr.shape) 
>>> reshaped.shape 
(1, 32, 32, 1)

來源

2016-12-30 16:32:10 MSeifert

轉換ndarray形狀

回答

相關問題