python 写 csv 文件的两种方法

zhzhx

方法一： csv

[Python] 纯文本查看 复制代码

import csv
log_path = 'log/temp.csv'
file = open(log_path, 'a+', encoding='utf-8', newline='')
csv_writer = csv.writer(file)
csv_writer.writerow([f'Epoch', 'Accuracy'])
csv_writer.writerow(['1', '96.25%'])
file.close()

结果：


方法二：pandas

[Python] 纯文本查看 复制代码

import os
import pandas as pd

path = 'temp.csv'
# 一次写入一行
df = pd.DataFrame(data=[['1', '98.2789%']])
# 一次写入多行
df = pd.DataFrame(data=[['1', '98.2789%'], ['2', '76.2345%']])
# 解决追加模式写的表头重复问题
if not os.path.exists(path):
    df.to_csv(path, header=['epoch', 'accuracy'], index=False, mode='a')
else:
    df.to_csv(path, header=False, index=False, mode='a')

结果：

2513002960

只是写csv文件的话pandas太重了

jobs_steven

csv吗？不应该写xlsx吗？

ying656

jobs_steven 发表于 2023-2-9 16:42
csv吗？不应该写xlsx吗？

好几种库都能操作的

jffwoo

pip3 install xlrd   #读取Excel文件
pip3 install xlwt   #创建xls文件，不支持创建xlsx