Загрузить данные из CSV в массив NumPy - PullRequest
0 голосов
/ 13 июня 2018

Я пытаюсь загрузить данные в 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 голосов
/ 13 июня 2018

Это должно дать вам необходимый вывод.

import numpy as np
def load_data(path):
    return np.loadtxt(path,delimiter=',')
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...