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

方法一：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为了换行更方便阅读  

方法二： numpy

    #导入包import pandas as pd
    import numpy as np
     
    df = pd.DataFrame(np.random.randn(10,4))#创建随机值
     
    #print(df.head(2))#查看数据框的头部数据，默认不写为前5行，小于5行时全部显示；也可以自定义查看几行
    print(df.tail())##查看数据框的尾部数据，默认不写为倒数5行，小于5行时全部显示；也可以自定义查看倒数几行
     
    df.to_csv('D:/PythonWorkSpace/TestData/PandasNumpy.csv')#存储到CSV中
    #df.to_excel('D:/PythonWorkSpace/TestData/PandasNumpy.xlsx')#存储到Excel中(需要提前导入库 pip install openpyxl)

方法三：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

可能的问题：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')