日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

分享python?寫?csv?文件的兩種方法_python

作者:全幼兒園最聰明 ? 更新時間: 2022-06-21 編程語言

問題描述

在深度學習相關任務的訓練時,需要在訓練的每個 epoch 記錄當前 epoch 的準確率(如下圖所示),那么在 python 中要怎么將內容寫入 csv 文件呢,學習發現可以使用 csv 或者 pandas 實現,在這里做個簡單記錄。

這里示例的代碼為以追加模式寫,每次寫入一行

方法一: csv

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

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')

執行完打開文件如下圖所示:

原文鏈接:https://blog.csdn.net/weixin_44589540/article/details/124369614

欄目分類
最近更新