Функция savetxt

Функция savetxt сохраняет массив NumPy в текстовый файл. Первым параметром указывается имя файла или файловый объект, вторым - сохраняемый массив. Опционально можно задать разделитель между элементами, формат вывода чисел, заголовок и комментарии. Функция полезна для экспорта данных в форматы CSV и другие текстовые форматы.

Синтаксис

np.savetxt(fname, X, delimiter=' ', fmt='%.18e', header='', footer='', comments='# ')

Параметры

Основные параметры функции savetxt:

fname - имя файла или файловый объект для записи
X - сохраняемый массив (одномерный или двумерный)
delimiter - строка-разделитель между элементами (по умолчанию пробел)
fmt - формат вывода чисел (строка или список строк)
header - строка, записываемая в начало файла
footer - строка, записываемая в конец файла
comments - строка, добавляемая перед заголовком и комментариями

Пример

Сохраним одномерный массив в текстовый файл с разделителем-запятой:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.savetxt('arr.txt', arr, delimiter=',')

Содержимое файла arr.txt:

1.000000000000000000e+00
2.000000000000000000e+00
3.000000000000000000e+00
4.000000000000000000e+00
5.000000000000000000e+00

Пример

Сохраним двумерный массив, используя формат с двумя знаками после запятой:

import numpy as np

arr = np.array([[1.123, 2.456, 3.789], [4.012, 5.345, 6.678]])
np.savetxt('arr.txt', arr, delimiter=',', fmt='%.2f')

Содержимое файла arr.txt:

1.12,2.46,3.79
4.01,5.35,6.68

Пример

Добавим заголовок и комментарии к сохраняемым данным:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
np.savetxt('arr.txt', arr, delimiter=',', fmt='%d',
		   header='Array data', comments='# ')

Содержимое файла arr.txt:

# Array data
1,2,3
4,5,6

Пример

Сохраним массив целых чисел с пользовательским форматом для каждого столбца:

import numpy as np

arr = np.array([[10, 20.5], [30, 40.7], [50, 60.2]])
np.savetxt('arr.txt', arr, delimiter='\t', fmt=['%d', '%.1f'])

Содержимое файла arr.txt:

10	20.5
30	40.7
50	60.2

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

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