CSVファイルに書き込んだデータを読み込みたい

Question

現在、CSVファイルへの書き込みと読み込みの勉強をしています。参考にしているURLが以下の通りです。
リンク内容
また、作成したコードは以下の通りです。

import csv
file = open("time.csv","a")
fileData = []
in_time = {"in_flag":True,"t1": 14.5686546547}
fileData.append(in_time)
csv.writer(file).writerow(fileData)

data = []
with open("time.csv","rb") as f:
    reader = csv.DictReader(f)
    for row in reader:
        data.append(row)
print(data)

一度CSVファイルに辞書を保存し、保存したCSVファイル内の辞書をdata内に復元しようとしているのですが、data内に値が入れられず出力は[]となってしまいます。私はこれまでCSVファイルを扱ったことがなかったため解決方法がわからない状況です。改善策を教えていただけると助かります。よろしくお願いします。

Answer 1

出力が空になる原因はfile.close()が呼び出されていないのが原因です。
スクリプトの終了時にファイルが自動的にクローズされて(?)いるのでただしく書き込めているように見えますが、
with open("time.csv","rb") as f: を呼ぶタイミングではまだ書き込みが反映されておらずファイルが空の状態で読み込まれるので当然出力も空になります。

import csv

file = open("time.csv", "a")
fileData = []
in_time = {"in_flag": True, "t1": 14.5686546547}
# fileData.append(in_time)
# csv.writer(file).writerow(fileData)
fields = ["in_flag", "t1"]
dw = csv.DictWriter(file, fieldnames=fields)
dw.writeheader()
dw.writerow(in_time)
file.close()


data = []
with open("time.csv", "r") as f:
    reader = csv.DictReader(f)
    for row in reader:
        data.append(row)
    print(data)

Answer 2

書き込むデータ fileData は dict のリストなので、DictWriter をお使いください。
読み込む際は、DictReader を使います。
for で1行ずつ読み取らなくても、data = list(reader) で一度に読み込めます。読み込んだデータも dict のリストになっています。

Python - CSV の読み込み/書き込み まとめ

import csv

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

with open("time.csv", "w") as f:
    # fieldnames 引数に列の順番指定
    writer = csv.DictWriter(f, fieldnames=['in_flag', 't1'])
    writer.writeheader()  # ヘッダーを書き込む。
    writer.writerows(data)  # データを書き込む。

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

flag = data['in_flag']
t1 = data['t1']

if flag:
    print('True のときの処理')
else:
    print('False のときの処理')

---

不明点等あれば、コメントしてください。

Comments

[Madao_0724]

回答ありがとうございます。tiitoi様のコードを参考に書き換えてみたところ、timeファイルの内容を読み込むことができました。読み込んだファイルから今回であればin_flagを変数に入れ、in_flagを入れた変数がTrueかFalseかを判断し処理分岐させるというようなことをやろうと考えています。また、t1も同様で、t1キーを参照して関連付けられた数値を変数に入れ、処理できるようにしたいと考えています。つまり、
data = {'in_flag':True,"t1":14534653}
a = data["in_flag"]
b = data["t1"]
という形にしたいと考えています。
上記のような処理をすることは無理なのでしょうか。回答していただけると助かります。長文となり申し訳ありません。

[tiitoi]

可能ですよ。ちなみにですが、書き込む際に "a" と追記モードを指定されてますが、これは意図した処理でしょうか？
つまり、in_flag、t1 のデータが何行も追記される形なのでしょうか？

[tiitoi]

in_flag と t1 の値が1つずつ保存されていればよいということであれば、'a' でなく、'w' フラグでファイルを書き込んでください。
質問のコードを変更しましたが、このような意図であっていますか？

[Madao_0724]

回答ありがとうございます。tiitoi様のコードを参考にしたところ、各キーの要素を変数に入れることができ、また型も自分が使いたいものに変換することができました。本当にありがとうございます。