将数据从csv加载到numpy数组

时间:2018-06-13 07:08:05

标签: python csv numpy

我正在尝试将csv文件(带分隔符',')中的数据加载到numpy数组中。一行的例子是:81905.75578271,81906.6205052,50685.487931,....(1000列)。 我有这个代码,但它似乎无法正常工作,因为在函数的退出调试器无法识别数据,当我调用xtrain.shape时,它返回0:

def load_data(path):
    # return np.loadtxt(path,dtype=int,delimiter=',')
    file = open(path,'r')
    data = []
    for line in file:
        array_vals = line.split(",")
        array = []
        for val in array_vals:
            if not val:
                array.append(float(val))
        data.append(np.asarray(array))
    return np.asarray(data)

x_train =  load_data(path)

1 个答案:

答案 0 :(得分:4)

这可以为您提供所需的输出。

import numpy as np
def load_data(path):
    return np.loadtxt(path,delimiter=',')