您可以在下面找到我用来将 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
我没有正确排序数据。你能请他帮我一下吗?提前致谢。
最佳答案
您可以使用 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
关于python - 将 sklearn 预测导出到 CSV 文件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52604751/