csvファイルの書き込みについて

Question

現在、ループ処理を行いシステムが動いている間欲しい値をCSVファイルに記録するプログラムを作成しています。以下がそのシステムの書き込み部分です。

with open(str(name) + ".csv","a") as f:
    writer = csv.DictWriter(f, fieldnames=['A', 'B','C'])
    writer.writeheader()  
    writer.writerows(total)

上記のコードをループ処理内で使用すると、数値と数値の間にヘッダーである「A」、「B」、「C」が記録されてしまいます。そこで、try文を使用し、ファイルが無かった場合のみ上記の処理をするようにしようと考え書き直したコードが以下に示す通りです。

try:
    with open(str(name) + ".csv","a") as f:
        writer.writerows(total) 
except:
    with open(str(name) + ".csv","w") as f:
        print("ファイルを作成します")
        writer = csv.DictWriter(f, fieldnames=['A', 'B','C'])
        writer.writeheader() 
        writer.writerows(total)

上記のコードをループ内に組み込んでプログラムを実行したところ、ファイルがあった場合でもexcept処理に入ってしまい、結局問題は解決しませんでした。最初だけヘッダーを付け、以降は数値だけを記録したい場合、どのようにコードを書き換えればよいか教えていただけると助かります。

Comments

[1T2R3M4]

解答がついたので削除

Answer 1

time.csv　が存在しない場合は、まずは w モードで open してヘッダだけのファイルをつくるようにします。
データの書き込みは常に a モードで open して書き込むようにします。

c.py

import csv
from pathlib import Path

csv_path = "time.csv"
headers = ["in_flag", "t1"]
data = []

in_time = {"in_flag": True, "t1": 14.5686546547}
data.append(in_time)

# ファイルが存在していなければ、まずはヘッダーのファイルを作成する
if not Path(csv_path).exists():
    with open(csv_path, "w") as f:
        csv.DictWriter(f, fieldnames=headers).writeheader()

# データを追記する
with open(csv_path, "a") as f:
    csv.DictWriter(f, fieldnames=headers).writerows(data)

with open(csv_path) as f:
    reader = csv.DictReader(f)
    data = list(reader)  # データを読み込む。

print(data)

実行例

Answer 2

最初にファイルの存在を判断し、ファイルが存在しなければ、新規ファイル作成してヘッダを書き込む、という処理にしましょう。

Answer 3

追記モード "a" を指定した場合、ファイルがない場合は作成するので、例外は発生しません。
pathlib の Path(ファイルパス).exists() でファイルが存在するかどうか調べて、存在しない場合はヘッダーを書き込むようにしてください。

import csv
from pathlib import Path

csv_path = "time.csv"

data = []
in_time = {"in_flag": True, "t1": 14.5686546547}
data.append(in_time)

exists = Path(csv_path).exists()

with open(csv_path, "a") as f:
    writer = csv.DictWriter(f, fieldnames=["in_flag", "t1"])
    if not exists:
        writer.writeheader()  # ファイルが存在しない場合は、ヘッダーを書き込む。
    writer.writerows(data)

with open(csv_path) as f:
    reader = csv.DictReader(f)
    data = list(reader)  # データを読み込む。

print(data)

Comments

[Madao_0724]

今回も助けていただきありがとうございます。各モードの使い方もしっかり学習しようと思います。ありがとうございます。