矢量化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)

的数组