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

---

掃碼關注公眾號：【哥妞】

個人博客：http://www.geniu.net/