三種方式使用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
)
|
這種代碼最少，最方便