将 sklearn 预测导出到 CSV 文件

Question

您可以在下面找到我用来将 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

我没有正确排序数据。你能请他让我出去吗？提前致谢。

Answer 1

您可以使用 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