Python Numpy – Сохранить массив в файл и прочитанный массив из файла

Автор оригинала: Python Examples.

Сохранить Numpy Array в файл и прочитайте Numpy Array из файла

Вы можете сохранить Numpy Array в файл, используя numpy.save (), а затем позже загрузите в массив, используя numpy.load ().

Ниже приведен фрагмент быстрого кода, в котором мы используем первое использование функции Save () для записи массива в файл. Во-вторых, мы используем нагрузку () функцию, чтобы загрузить файл в Numpy Array.

# save array to file
numpy.save(file, array)
# load file to array
array = numpy.load(file)

Пример 1: Сохраните массив в файл

В следующем примере: мы инициализируем массив; Создайте и откройте файл в записи двоичного режима; А затем напишите массив в файл, используя mempy.save () метод.

Python Program

import numpy as np

#initialize an array
arr = np.array([[[11, 11, 9, 9],
                  [11, 0, 2, 0]
				 ],
	             [[10, 14, 9, 14],
                  [0, 1, 11, 11]]])

# open a binary file in write mode
file = open("arr", "wb")
# save array to the file
np.save(file, arr)
# close the file
file.close

Как только вы закончите сохранение массива, не забудьте закрыть файл.

Вы должны соблюдать, что новый файл с именем arr создается в вашем текущем рабочем каталоге. Расширение файла не требуется. Но вы можете использовать расширение по вашему выбору.

Пример 2: Загрузите сохраненный насыпчатый массив из файла

В этом примере мы загрузим массив из файла. Мы будем использовать вышеприведенный пример, чтобы сохранить массив и продолжить, что читать массив из файла.

Python Program

import numpy as np

#initialize an array
arr = np.array([[[11, 11, 9, 9],
                  [11, 0, 2, 0]
				 ],
	             [[10, 14, 9, 14],
                  [0, 1, 11, 11]]])

# open a binary file in write mode
file = open("arr", "wb")
# save array to the file
np.save(file, arr)
# close the file
file.close

# open the file in read binary mode
file = open("arr", "rb")
#read the file to numpy array
arr1 = np.load(file)
#close the file
print(arr1)

Выход

[[[11 11  9  9]
  [11  0  2  0]]

 [[10 14  9 14]
  [ 0  1 11 11]]]

Мы успешно прочитали MAMPY MARY из файла и загрузить объект с этим массивом.

Резюме

В этом Numpy учебник примеров Python мы изучали, как сохранить Numpy Array в файл, и загрузить Numpy Array из файла в объект в программе.