Python写入数据到csv或xlsx文件的3种方法
本文介绍Python中三种将数据写入CSV或Excel文件的方法:使用csv模块处理CSV文件,利用openpyxl模块操作xlsx格式,以及通过pandas库同时支持两种格式。其中pandas方法代码最简洁,适合快速实现数据持久化需求。
Python写入CSV或Excel文件的三种方法
本文介绍三种使用Python将数据写入CSV或Excel文件的方法,供读者参考。
使用csv模块写入CSV文件
第一种方法:使用csv模块,将数据写入CSV格式文件。
# -*- coding: utf-8 -*-
import csv
with open("my.csv", "a", newline='') as f:
writer = csv.writer(f)
writer.writerow(["URL", "predict", "score"])
row = [['1', 1, 1], ['2', 2, 2], ['3', 3, 3]]
for r in row:
writer.writerow(r)
使用openpyxl模块写入XLSX文件
第二种方法:使用openpyxl模块,将数据写入XLSX格式文件。
# -*- coding: utf-8 -*-
import openpyxl as xl
import os
def write_excel_file(folder_path):
result_path = os.path.join(folder_path, "my.xlsx")
print(result_path)
print('***** 开始写入excel文件 ' + result_path + ' ***** \n')
if os.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]]
for data in result:
sheet.append(data)
workbook.save(result_path)
print('***** 生成Excel文件 ' + result_path + ' ***** \n')
if __name__ == '__main__':
write_excel_file("D:\core\\")
使用pandas库写入文件
第三种方法:使用pandas库,该库支持写入CSV或XLSX格式文件。
import pandas as pd
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)
这种方法代码量最少,操作最为便捷。
以上为本文全部内容,希望能为读者提供有益参考。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









