我有一个形状为(20,50,100,500,500)
的numpy数组,并且我想根据第3维(例如40/60)对数组进行切片。
我所能想到的就是要做。 array[:,:,:40,:,:]
和array[:,:,60:,:,:]
,但是如何在不弄乱尺寸的情况下将它们连接起来呢?
答案 0 :(得分:1)
我使用了np.stack,将df_inventory = inventory.reset_index().rename(columns={'col_name':'desired_name'})
设置如下:
axis=2
希望这会有所帮助。
答案 1 :(得分:0)
仅在右轴上串联:
np.concatenate([ array[:,:,:40,:,:], array[:,:,60:,:,:] ], axis=2)