Функция load

Функция load загружает массивы NumPy из файлов в бинарных форматах .npy и .npz, а также из текстовых файлов. Первым параметром функция принимает путь к файлу или файловый объект. Вторым параметром можно указать режим отображения данных с помощью параметра mmap_mode.

Синтаксис

np.load(file, mmap_mode=None, allow_pickle=True,
	fix_imports=True, encoding='ASCII')

Пример

Давайте сохраним массив в бинарный файл и загрузим его обратно:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.save('data.npy', arr)

loaded_arr = np.load('data.npy')
print(loaded_arr)

Результат выполнения кода:

[1 2 3 4 5]

Пример

Загрузим несколько массивов из файла .npz:

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
np.savez('data.npz', a=arr1, b=arr2)

data = np.load('data.npz')
print(data['a'])
print(data['b'])

Результат выполнения кода:

[1 2 3]
[4 5 6]

Пример

Загрузим данные из текстового файла с помощью load:

1 2 3
4 5 6

import numpy as np

arr = np.load('data.txt')
print(arr)

Результат выполнения кода:

[[1 2 3]
 [4 5 6]]

Пример

Загрузим массив в режиме отображения в памяти для работы с большими данными:

import numpy as np

np.save('large_data.npy', np.arange(1000))
mmap_arr = np.load('large_data.npy', mmap_mode='r')
print(mmap_arr[0])
print(mmap_arr[999])

Результат выполнения кода:

0
999

Смотрите также

функцию save,
которая сохраняет массив в бинарный файл .npy
функцию savez,
которая сохраняет несколько массивов в один файл .npz
функцию loadtxt,
которая загружает данные из текстового файла с возможностью указания разделителя
функцию genfromtxt,
которая загружает данные с поддержкой пропущенных значений