首页 > 编程 > Python > 正文

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

2019-11-25 11:52:59
字体:
来源:转载
供稿:网友

本文实例为大家分享了三种方式使用python写数据到csv或xlsx文件,供大家参考,具体内容如下

第一种:使用csv模块,写入到csv格式文件

# -*- coding: utf-8 -*-import csvwith 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格式文件

# -*- coding: utf-8 -*-import openpyxl as xlimport osdef 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格式文件

import pandas as pdresult_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)

这种代码最少,最方便

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持武林网。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表