python写入数据到csv或xlsx文件的3种方法
本文实例为大家分享了三种方式使用python写数据到csv或xlsx文件,供大家参考,具体内容如下
第一种:使用csv模块,写入到csv格式文件
#-*-coding:utf-8-*-
importcsv
withopen("my.csv","a",newline='')asf:
writer=csv.writer(f)
writer.writerow(["URL","predict","score"])
row=[['1',1,1],['2',2,2],['3',3,3]]
forrinrow:
writer.writerow(r)
第二种:使用openpyxl模块,写入到xlsx格式文件
#-*-coding:utf-8-*-
importopenpyxlasxl
importos
defwrite_excel_file(folder_path):
result_path=os.path.join(folder_path,"my.xlsx")
print(result_path)
print('*****开始写入excel文件'+result_path+'*****\n')
ifos.path.exists(result_path):
print('*****excel已存在,在表后添加数据'+result_path+'*****\n')
workbook=xl.load_workbook(result_path)
else:
print('*****excel不存在,创建excel'+result_path+'*****\n')
workbook=xl.Workbook()
workbook.save(result_path)
sheet=workbook.active
headers=["URL","predict","score"]
sheet.append(headers)
result=[['1',1,1],['2',2,2],['3',3,3]]
fordatainresult:
sheet.append(data)
workbook.save(result_path)
print('*****生成Excel文件'+result_path+'*****\n')
if__name__=='__main__':
write_excel_file("D:\core\\")
第三种,使用pandas,可以写入到csv或者xlsx格式文件
importpandasaspd
result_list=[['1',1,1],['2',2,2],['3',3,3]]
columns=["URL","predict","score"]
dt=pd.DataFrame(result_list,columns=columns)
dt.to_excel("result_xlsx.xlsx",index=0)
dt.to_csv("result_csv.csv",index=0)
这种代码最少,最方便
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。