Экспорт прогнозов sklearn в файл CSV - PullRequest
Новая
       8

Экспорт прогнозов sklearn в файл CSV

0 голосов
/ 02 октября 2018

Ниже приведен код, который я использую для экспорта прогнозов sklearn в файл CSV.Как видите, я делаю прогнозы из CSV.Все работает нормально, но вывод, который я получаю с возвращенным CSV-файлом, не отображает данные правильно:

Часть кода: 

path = "C:\\Users\\USER\\Desktop\\testimport.csv"
file = open(path, newline='')
reader = csv.reader(file)

header = next(reader) #la 1ere ligne correspond au titre
data = []
id_trajet =  []
for row in reader:

    track_id  = int(row[0])
    duree =  float(row[1])
    distance = float(row[2])
    vr_gauche = float(row[3])
    vr_droite = float(row[4])
    acc = float(row[5])
    freinages = float(row[6])
    data.append([duree, distance, vr_gauche, vr_droite, acc, freinages])
    id_trajet.append(track_id)


prediction = clf.predict (data)

#print(prediction)

returns_path = "C:\\Users\\USER\\Desktop\\testexport.csv"
file = open(returns_path, 'w', newline='')
writer = csv.writer(file, delimiter=',')
writer.writerow(["Id_Trajet", "Type_Conduite"])
writer.writerow(id_trajet)
writer.writerow(prediction)


print(prediction)

Что я получаю прямо сейчас с CSVфайл: 

Id_Trajet,Type_Conduite
357,666
intermediaire,sportive

Что я хотел бы получить из файла CSV: 

Id_Trajet,Type_Conduite
357,intermediaire
666,sportive

Я неправильно сортирую данные.Не могли бы вы доставить ему удовольствие?Заранее спасибо.

1 Ответ

0 голосов
/ 02 октября 2018

Вы можете использовать Pandas DataFrame для обработки данных и экспорта. 

import pandas

id_trajet = [357, 666]
prediction = ['intermediaire', 'sportive']

df = pandas.DataFrame(prediction, columns=['Type_Conduite'], index=id_trajet)
df.index.name = 'Id_Trajet'

print(df.to_csv())

вывод: 

Id_Trajet,Type_Conduite
357,intermediaire
666,sportive
...