Вы можете сохранить массив numpy в файл с помощью save(), а затем загрузить в массив с помощью load().
Ниже приведен небольшой фрагмент кода, в котором мы сначала используем функцию save() для записи массива в файл. Во-вторых, мы используем функцию load() для загрузки файла в массив numpy.
# save array to file numpy.save(file, array) # load file to array array = numpy.load(file)
Пример 1
В следующем примере мы инициализируем массив, создадим и откроем файл в двоичном режиме записи, а затем запишем массив в файл с помощью метода numpy.save().
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
В этом примере мы загрузим массив из файла. Мы будем использовать приведенный выше пример для сохранения массива и продолжения чтения массива из файла.
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]]]
Мы успешно прочитали массив numpy из файла и загрузили объект с этим массивом.
Заключение
В этом учебном пособии по примерам Python мы узнали, как сохранить массив numpy в файл и загрузить из файла в объект в программе.