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

Dara ·

更新时间:2024-11-14

· 842 次阅读

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

第一种：使用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格式文件


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


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)

这种代码最少，最方便

您可能感兴趣的文章:python3+selenium实现qq邮箱登陆并发送邮件功能python3+selenium实现126邮箱登陆并发送邮件功能实例分析python3实现并发访问水平切分表python自动查询12306余票并发送邮箱提醒脚本Python3实现并发检验代理池地址的方法python配置文件写入过程详解python config文件的读写操作示例python批量读取文件名并写入txt文件中Python3并发写文件与Python对比

xlsx 方法数据 csv Python

1024 个赞