如何高效将(7201,)的1D NumPy数组转换为(X,10,1)的3D数组?
将嵌套数组转换为形状(X,10,1)的高效方法
你的x_train是一个包含7201个(10,1)形状子数组的一维数组,要转成目标形状的三维数组,用numpy内置函数就能高效实现,以下是两种最直接的方案:
方案一:直接用
np.array()转换
因为所有子数组形状完全一致,直接将整个嵌套数组传入np.array()就能自动堆叠成三维数组:import numpy as np x_train_reshaped = np.array(x_train) # 输出形状:(7201, 10, 1) print(x_train_reshaped.shape)这是效率最高的方式,numpy底层用C实现内存分配和数据复制,比任何Python循环都快。
方案二:用
np.stack()堆叠
如果直接转换出现异常(比如子数组内存不连续),可以用np.stack()指定轴来堆叠:x_train_reshaped = np.stack(x_train, axis=0) print(x_train_reshaped.shape) # 同样输出(7201, 10, 1)这个方法逻辑更直观,把每个子数组作为一个"块"沿0轴堆叠,性能和
np.array()几乎无差别。
注意:两种方法都要求所有子数组形状必须是(10,1),有形状不一致的子数组会报错,你的案例显然符合条件。
内容的提问来源于stack exchange,提问作者cocktail
相关产品推荐
相关产品推荐

