删除N-D numpy数组的第一个元素

时间:2018-01-12 02:34:34

标签: python arrays numpy

我有一个numpy 5D数组,想要删除第一个元素并将一个新元素推送到索引的末尾。这是定义;

images = np.zeros([1,30,image_height,image_width,image_channel])

这是要删除的内容:images[0,0,:,:,:]

这是要更新的内容:images[0,29,:,:,:]

如何从N-D大小的数组中删除和添加元素?

谢谢,

1 个答案:

答案 0 :(得分:1)

好的,我发现了。

images[0, 0:28, :, :, :] = images[0, 1:29, :, :, :]

images[0, 30, :, :, :] = new_data

要避免硬编码索引,可将其写为:

images[0,0:-1,...] = images[0,1:,...]
images[0,-1,...] = new_data