vectorize 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

あなたはあなたが必要とするデータを抽出するためにスライスを使用することができます形状(10, 1, 96)の配列が得られます

ff[:10,:,:] 

。空の次元を取り除くために、あなたはnumpy.squeeze()を通してそれを実行することができます。

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

をし、形状(10, 96)