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
データを正しく並べ替えません。彼に私を喜ばせてくれませんか。前もって感謝します。
PandasDataFrameを使用してデータの操作とエクスポートを行うことができます。
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
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加