哥妞 海淀山后有哥妞

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

2020-03-26

几种保存数据到text、csv文件的方法。

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

方法一:open函数保存

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

方法二:csv写入

import csv
import 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')

1分也是爱


微信扫描二维码,关注公众号 (转载本站文章请注明作者和出处 哥妞-geniu)

评论

Content