三种方式使用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)

这种代码最少，最方便

字典格式写法

data = {“URL”: [‘1’, 1, 1], “predict”: [‘2’, 2, 2], “score”: [‘3’, 3, 3]}
d = pd.DataFrame(data)
d.to_excel(“result.xlsx”, index=0)

如果
df = pd.DataFrame({‘A’:0})这样写，会报错
ValueError: If using all scalar values, you must pass an index