python保存数据到文件的几种方法

在使用python训练人工智能网络模型的时候，经常需要保存训练结果到文件中，比如在kaggle竞赛的时候，通常需要保存测试结果到csv中，这个时候大家可能会用到本文的技术。

方法一：open函数保存

    #保存数据 open函数    with open('D:/PythonWorkSpace/TestData/pinglun.txt','w',encoding='utf-8') as f:        for i in comments:            print(i)            # 入数据，文件保存在上面指定的目录，加\n为了换行更方便阅读            f.write(i+'\n')

方法二：csv写入

import csvimport codecs
with codecs.open('./test.csv', 'w', 'utf-8') as csvfile:    # 指定 csv 文件的头部显示项    filednames = ['ID', 'PRICE']    writer = csv.DictWriter(csvfile, fieldnames=filednames)    writer.writeheader()    for i in range(0, len(test_index)):        try:            writer.writerow({'ID':test_index[i], 'PRICE':y_pred[i]})        except UnicodeEncodeError:            print("编码错误, 该数据无法写到文件中, 直接忽略该数据")

    

方法三：DataFrame

使用 DataFrame 包，这也是 pandas 中训练人工智能模型常用的 python 包，是一个大家都必会的包。

可能的问题：csv文件中看不到数据，但是通过python代码可以看到数据。   

dataframe = pd.DataFrame({'ID':test_index,'PRICE': y_pred})# dataframe = pd.DataFrame({'PRICE': test_index})dataframe.to_csv("test12.csv",index=False,sep='\n')

本文分享自微信公众号 - 哥妞（gh_d18ec82f19ea）。
如有侵权，请联系 [email protected] 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一起分享。