作者：布丁宝宝-_932 | 来源：互联网 | 2023-02-01 17:55
------------恢复内容开始------------三种方式使用python写数据到csv或xlsx文件：第一种：使用csv模块，写入到csv格式文件123456789#-
------------恢复内容开始------------
三种方式使用python写数据到csv或xlsx文件：
第一种：使用csv模块，写入到csv格式文件
1
2
3
4
5
6
7
8
9
|
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
|
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)
|
这种代码最少，最方便
------------恢复内容结束------------
京公网安备 11010802041100号