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