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/