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

将sklearn预测导出到CSV文件

1 个答案: