Як експортувати масив 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
n.p. 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
n.p. 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)
n.p. 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)
n.p. savetxt (" my_data.csv ", data, delimiter=" , ", fmt=" %.2f ",
           header=" A, B, C ", comments="")

Примітка . Аргумент коментарів запобігає відображенню символу «#» у заголовках.

Якщо я перейду туди, де збережено файл CSV, я зможу переглянути дані:

Примітка . Ви можете знайти повну документацію для функції numpy.savetxt() тут .

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції читання та запису в Python:

Як читати файли CSV за допомогою NumPy
Як читати файли CSV за допомогою Pandas
Як експортувати Pandas DataFrame у файл CSV