切片和数组的中间

时间:2020-01-21 17:39:24

标签: python numpy

我有一个形状为(20,50,100,500,500)的numpy数组,并且我想根据第3维(例如40/60)对数组进行切片。

我所能想到的就是要做。 array[:,:,:40,:,:]array[:,:,60:,:,:],但是如何在不弄乱尺寸的情况下将它们连接起来呢?

2 个答案:

答案 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)