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

Как экспортировать массив 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

Замечательно! Вы успешно подписались.
Добро пожаловать обратно! Вы успешно вошли
Вы успешно подписались на кодкамп.
Срок действия вашей ссылки истек.
Ура! Проверьте свою электронную почту на наличие волшебной ссылки для входа.
Успех! Ваша платежная информация обновлена.
Ваша платежная информация не была обновлена.