Как выполнить чтение и запись массивов NumPy в файлы?

Пользователь

от greyson , в категории: Python , 2 месяца назад

Как выполнить чтение и запись массивов NumPy в файлы?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от chloe.keebler , 2 месяца назад

@greyson 

Чтение и запись массивов NumPy в файлы можно выполнить с помощью функций numpy.save и numpy.load.


numpy.save позволяет сохранить один или несколько массивов NumPy в бинарный файл формата .npy, в то время как numpy.load загружает массивы из таких файлов.


Примеры использования функций numpy.save и numpy.load:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
import numpy as np

# Создаем массивы
arr1 = np.array([1, 2, 3])
arr2 = np.array([[1, 2, 3], [4, 5, 6]])

# Сохраняем массивы в файл
np.save('arr1.npy', arr1)
np.save('arr2.npy', arr2)

# Загружаем массивы из файла
loaded_arr1 = np.load('arr1.npy')
loaded_arr2 = np.load('arr2.npy')

print(loaded_arr1) # [1 2 3]
print(loaded_arr2) # [[1 2 3]
                   #  [4 5 6]]


Также можно сохранять несколько массивов в один файл с помощью функции numpy.savez, которая создает сжатый архив .npz.


Пример использования функции numpy.savez:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
import numpy as np

# Создаем массивы
arr1 = np.array([1, 2, 3])
arr2 = np.array([[1, 2, 3], [4, 5, 6]])

# Сохраняем массивы в один файл
np.savez('arrays.npz', arr1=arr1, arr2=arr2)

# Загружаем массивы из файла
loaded_data = np.load('arrays.npz')
loaded_arr1 = loaded_data['arr1']
loaded_arr2 = loaded_data['arr2']

print(loaded_arr1) # [1 2 3]
print(loaded_arr2) # [[1 2 3]
                   #  [4 5 6]]


При загрузке массивов из файла можно использовать и другие форматы, например, текстовый формат с помощью функций numpy.savetxt и numpy.loadtxt.