Вы можете использовать следующий базовый синтаксис для экспорта массива NumPy в файл CSV:
import numpy as np
#define NumPy array
data = np.array([[1,2,3],[4,5,6],[7,8,9]])
#export array to CSV file
np.savetxt (" my_data.csv", data, delimiter=" , ")
В следующих примерах показано, как использовать этот синтаксис на практике.
Пример 1: экспорт массива NumPy в CSV
В следующем коде показано, как экспортировать массив NumPy в файл CSV:
import numpy as np
#define NumPy array
data = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])
#export array to CSV file
np.savetxt (" my_data.csv", data, delimiter=" , ")
Если я перейду к месту, где файл CSV сохранен на моем ноутбуке, я смогу просмотреть данные:
Пример 2: экспорт массива NumPy в CSV в определенном формате
Формат чисел по умолчанию — «%.18e» — это отображает 18 нулей. Однако мы можем использовать аргумент fmt для указания другого формата.
Например, следующий код экспортирует массив NumPy в CSV и указывает два десятичных разряда:
import numpy as np
#define NumPy array
data = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])
#export array to CSV file (using 2 decimal places)
np.savetxt (" my_data.csv", data, delimiter=" ,", fmt=" %.2f ")
Если я перейду к месту, где сохранен файл CSV, я смогу просмотреть данные:
Пример 3: экспорт массива NumPy в CSV с заголовками
В следующем коде показано, как экспортировать массив NumPy в файл CSV с пользовательскими заголовками столбцов:
import numpy as np
#define NumPy array
data = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])
#export array to CSV file (using 2 decimal places)
np.savetxt (" my_data.csv", data, delimiter=" ,", fmt=" %.2f",
header=" A, B, C", comments="")
Примечание.Аргумент комментариев предотвращает отображение символа «#» в заголовках.
Если я перейду к месту, где сохранен файл CSV, я смогу просмотреть данные:
Примечание.Полную документацию по функции numpy.savetxt() можно найти здесь .
Дополнительные ресурсы
В следующих руководствах объясняется, как выполнять другие распространенные операции чтения и записи в Python:
Как читать файлы CSV с помощью NumPy
Как читать файлы CSV с помощью Pandas
Как экспортировать Pandas DataFrame в файл CSV