我正在尝试对以下循环进行矢量化，我正在尝试将多个数组附加到一个空数组。
# ff is a matrix of shape [100,1,96]
temp = np.array([]).reshape(0,96)
for kk in range(1,10,1):
temp = np.append(tr,ff[kk],axis=0)
temp = temp.reshape(1,10,96)
是否可以使用 numpy 对上述循环进行矢量化？欢迎任何帮助!
请您参考如下方法:
您可以使用切片来提取您需要的数据:
ff[:10,:,:]
这将产生一个形状为 (10, 1, 96) 的数组。要摆脱空维度，您可以通过 numpy.squeeze() 运行它:
numpy.squeeze(ff[:10,:,:])
并得到一个形状为(10, 96)的数组
