Как извлечь часть массива numpy?

Question

Я пытаюсь извлечь часть массива, используя numpy. Я смотрю на это, и кажется, что есть много способов сделать это (например, np.delete () ). Однако мне не удается заставить его работать для моей проблемы.

У меня большой массив с shape(80,2,100,100,3) Я хочу разделить этот массив на две скользящие части, используя al oop:

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