转换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形状

回答

相关问题