我是編程新手，正在嘗試減少將數(shù)據(jù)寫(xiě)入文件所需的時(shí)間，但我發(fā)現(xiàn)寫(xiě)入部分是主要問(wèn)題。以下是我的機(jī)器學(xué)習(xí)程序代碼的一部分：filename="data.csv"f=open(filename,"w")headers="row,open\n"f.write(headers)for i in range (0,55970): score=rf.predict(edit[i].reshape(1, -1)) score=str(score).replace('[','').replace(']','') f.write(str(i) +","+ score +"\n") f.close()我知道我應(yīng)該只在獲得所有數(shù)據(jù)后才寫(xiě)入數(shù)據(jù)，但我不確定如何去做——因?yàn)槲抑恢?f.write()。我是否為我的預(yù)測(cè)和返回分?jǐn)?shù)創(chuàng)建一個(gè)函數(shù)，然后創(chuàng)建一個(gè)列表來(lái)存儲(chǔ)所有分?jǐn)?shù)并將其寫(xiě)入？（如果可能的話）[編輯]score=rf.predict(edit)with open('data.csv', 'w',newline='') as f: writer = csv.writer(f) writer.writerow(['row', 'open']) for i in range(55970): writer.writerow([i,str(score[i])])^ 根據(jù)新建議添加。發(fā)現(xiàn)我應(yīng)該只做預(yù)測(cè)，然后寫(xiě)下顯著縮短時(shí)間的行！感謝您的幫助?。?
1 回答
汪汪一只貓
TA貢獻(xiàn)1898條經(jīng)驗(yàn) 獲得超8個(gè)贊
CSV 模塊是一個(gè)更好的工具。更具體地說(shuō)，writerows()是您正在尋找的東西。
https://docs.python.org/3/library/csv.html#csv.csvwriter.writerows
這是文檔中的示例：
import csv
with open('some.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(someiterable)
import csv
with open('data.csv', 'w') as csv_file:
writer = csv.writer(csv_file)
writer.writerow(['row_id', 'open_flag'])
for i in range(55970):
score = str(rf.predict(edit[i].reshape(1, -1)))
score.replace('[', '').replace(']', '')
writer.writerow([i, score])
添加回答
舉報(bào)
0/150
提交
取消