如何提取部分数字数组?

Question

我正在尝试使用 numpy 提取数组的一部分。我查了一下，似乎也有很多方法可以做到这一点(例如 np.delete(((。但是，我无法让它解决我的问题。

我有一个带有shape(80,2,100,100,3)的大数组，我想使用循环将这个数组分成两个滑动部分：

for i in range(0, len(data), len(data)//2):
part_1 = data[i:i+len(data)//5] #Initial selection
part_2 = data[0:i::i+len(data)//5:] #Selection of the rest of the initial data
for i in range (0, len(data), len(data)//2):
part_1 = data[i:i+len(data)//5]
part_2 = np.delete(data, obj=part_1, axis=0)

这两种解决方案都不起作用。有人有想法吗？

亲切问候

Unic0

Answer 1

我找到了一种方法，通过将问题分为两部分来做到这一点：

for i in range(0,len(data),len(data)//5):
part_1 = data[i:i+len(data)//5]
temp0 = data[0:i]
temp1 = data[i+len(data)//5:]
part_2 = np.concatenate((train_images0, train_images1), axis=0)

但是，我确信会有更好的方法可以一步到位地做到这一点。

如果有人有任何建议，我们将非常欢迎。

问候

Unic0