Как экспортировать массив NumPy в файл CSV (с примерами)

Вы можете использовать следующий базовый синтаксис для экспорта массива 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