Как прочитать файл CSV с помощью NumPy (шаг за шагом)


Вы можете использовать следующий базовый синтаксис для чтения файла CSV в массив записей в NumPy:

from numpy import genfromtxt

my_data = genfromtxt('data.csv', delimiter=',', dtype= None )

В следующем пошаговом примере показано, как использовать этот синтаксис на практике.

Шаг 1. Просмотрите CSV-файл

Предположим, у нас есть следующий CSV-файл с именем data.csv , который мы хотим прочитать в NumPy:

Шаг 2. Чтение в CSV-файле

Следующий код показывает, как читать этот файл CSV в массив Numpy:

from numpy import genfromtxt

#import CSV file
my_data = genfromtxt('data.csv', delimiter=',', dtype= None )

Обратите внимание на следующее:

  • разделитель : указывает разделитель, который разделяет значения данных в CSV-файле.
  • dtype : указывает тип данных для массива NumPy. Используя None , мы разрешаем одновременно импортировать несколько типов данных в массив.

Пример 3: просмотр массива NumPy

После того, как мы импортировали файл CSV, мы можем просмотреть его:

#view imported CSV file
my_data

array([[1, 2, 2, 2, 3, 4],
 [5, 5, 6, 8, 9, 9]])

Мы видим, что данные в массиве NumPy соответствуют данным, показанным в файле CSV.

Примечание.Полную онлайн-документацию по функции genfromtxt() можно найти здесь .

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные функции с CSV-файлами в pandas:

Как читать файлы CSV с помощью Pandas
Как экспортировать Pandas DataFrame в файл CSV
Pandas: как добавить данные в существующий файл CSV