python - 矢量化 numpy 追加循环

Question

我正在尝试对以下循环进行矢量化，我正在尝试将多个数组附加到一个空数组。

# 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 对上述循环进行矢量化？欢迎任何帮助!

Answer 1

您可以使用切片来提取您需要的数据:

ff[:10,:,:]

这将产生一个形状为 (10, 1, 96) 的数组。要摆脱空维度，您可以通过 numpy.squeeze() 运行它:

numpy.squeeze(ff[:10,:,:])

并得到一个形状为(10, 96)的数组