Чтение файла CSV в массив Numpy

Question

Я довольно новичок в numpy. Я пытаюсь открыть файл CSV, который содержит 4 столбца с около 100 строк. Я хочу иметь двумерный массив, который устанавливает первый столбец как координаты x и устанавливает второй столбец как координаты y.

import numpy as np

dataDocument = open("data.csv")
headers = dataDoument.readline()

def generateArray(dataDocument):
    for numbers in dataDocument:
        splitDocument = numbers.strip().split(",")
        myArray = np.array(splitDocument[0], splitDocument[1])

        return myArray 

print(generateArray(dataDocument))

Я продолжаю получать различные сообщения об ошибках, наиболее распространенным из которых является «тип данных», «не понятый». Любые предложения о том, где моя логическая ошибка / общая ошибка кода существует?

Answer 1

Вы также можете попробовать:

 import numpy as np
 d = np.loadtxt('a.csv', delimiter=',')
 x = d[:,0]
 y = d[:,1]

Answer 2

Попробуйте:

from numpy import genfromtxt    
data = genfromtxt('data.csv', delimiter=',')

res = data[:,0:2]

Answer 3

replace: myArray = np.array(splitDocument[0], splitDocument[1])
with: myArray = np.array((splitDocument[0], splitDocument[1])) 
in your method and should solve the issue.

Проблема связана с тем, что вы передаете splitDocument[1] в np.array в качестве параметра dtype.