Сохранение массива numpy после индексации происходит намного медленнее ⇐ Python

[Anonymous]

Я столкнулся с проблемой: сохранение массива numpy после индексации приводит к гораздо более медленному сохранению. Минимальный воспроизводимый пример можно увидеть ниже:

Код: Выделить всё

import time
import numpy as np

def mre(save_path):
array = np.zeros((245, 233, 6))

start = time.time()
for i in range(1000):
with open(save_path + '/array1_' + str(i), "wb") as file:
np.save(file, array)
end = time.time()
print(f"No indexing: {end - start}s")

array2 = array[:,:,[0,1,2,3,4,5]]
start = time.time()
for i in range(1000):
with open(save_path + '/array2_' + str(i), "wb") as file:
np.save(file, array2)
end = time.time()
print(f"With indexing: {end - start}s")
print("Arrays are equal: ", np.array_equal(array, array2))

В результате:

Код: Выделить всё

No indexing: 2.9975574016571045s
With indexing: 10.408239126205444s
Arrays are equal:  True

Таким образом, согласно numpy массивы равны, но время сохранения в результате значительно медленнее. Есть ли у кого-нибудь идеи, почему это так?

Подробнее здесь: https://stackoverflow.com/questions/791 ... uch-slower