Запись массива NumPy в файл CSV

Часто перед аналитиками данных и исследователями встает задача сохранения результатов в формате, который удобен для чтения и дальнейшего использования. Одним из таких форматов является CSV (Comma-Separated Values) — текстовый формат, который используется для представления табличных данных. Каждая строка в таком файле — это строка таблицы, а столбцы разделяются запятой.

Рассмотрим типичный пример. Предположим, что был произведен расчет или анализ данных, и результаты этих действий хранятся в двумерном массиве NumPy. Теперь эти данные нужно сохранить в файле CSV для дальнейшего использования, например, для анализа в других программах.

Для записи массива NumPy в файл CSV можно использовать функцию numpy.savetxt. Эта функция сохраняет двумерный массив в текстовый файл. В качестве первого аргумента функция принимает имя файла, в который будет произведена запись. Вторым аргументом является массив, который нужно записать. Также можно указать разделитель (по умолчанию пробел), который будет использоваться для отделения значений в строке.

Пример записи массива в CSV-файл:

import numpy as np

# Создаем двумерный массив
array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Записываем массив в файл CSV
np.savetxt("array.csv", array, delimiter=",")

После выполнения этого кода в корневой директории проекта появится файл array.csv, содержащий данные из массива array. Если открыть файл, то увидим следующее:

1.000000000000000000e+00,2.000000000000000000e+00,3.000000000000000000e+00
4.000000000000000000e+00,5.000000000000000000e+00,6.000000000000000000e+00
7.000000000000000000e+00,8.000000000000000000e+00,9.000000000000000000e+00

Если требуется записать данные в более читаемом формате, можно использовать параметр fmt функции numpy.savetxt, который позволяет указать формат данных. Например, %d для целых чисел и %.2f для чисел с плавающей точкой с двумя знаками после запятой:

np.savetxt("array.csv", array, delimiter=",", fmt="%d")

Теперь в файле array.csv хранятся данные в следующем виде:

1,2,3
4,5,6
7,8,9

Итак, функция numpy.savetxt является удобным инструментом для сохранения двумерных массивов NumPy в файлы CSV.