python写入数据到csv或xlsx文件的3种方法

三种方式使用python写数据到csv或xlsx文件：

第一种：使用csv模块，写入到csv格式文件

1 2 3 4 5 6 7 8 9

# -*- 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格式文件

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28

# -*- 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，可以写入到csv或者xlsx格式文件

1 2 3 4 5 6

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)

这种代码最少，最方便