将 CSV 文件读入 Numpy 数组

Question

我正在尝试打开一个 CSV 文件，其中包含 4 列大约 100 行。 我想要一个二维 numpy 数组，它将第一列设置为 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

尝试：

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

res = data[:,0:2]

Answer 2

你也可以试试：

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

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参数传递给您的np.array 。