前提・実現したいこと
python学習中です(初投稿につき無礼がありましたら申し訳ありません)。
csvモジュールをいじっているのですが、readerの挙動で不明な点があります。
forを使って一行ずつ読み取っていき、
指定の行数に至ったときに別処理をする、ということを実現したいのですが、
readerとfor文の間にreaderを扱うコードを記載すると、
以降のfor文が動きません(エラーも出ずに、何の処理もせず終了する)。
調べてみたのですが原因が分からず、ご教示いただけると幸いです。
該当のソースコード
python
1 2import csv 3import os 4 5if os.path.exists("./csv_train.csv"): 6 pass 7 8else: 9 with open("csv_train.csv", "w", newline="") as csv_file: 10 fieldnames = ["NAME", "COUNT"] 11 writer = csv.DictWriter(csv_file, fieldnames=fieldnames) 12 writer.writeheader() 13 14 writer.writerow({"NAME": "aaa", "COUNT": 1}) 15 writer.writerow({"NAME": "aaa", "COUNT": 1}) 16 writer.writerow({"NAME": "aaa", "COUNT": 1}) 17 18with open("csv_train.csv", "r") as csv_file: 19 reader = csv.DictReader(csv_file) 20 21 #↓を無くすと、forが動く。この文があるとforが動かない。 22 cnt = len(list(reader)) 23 #↑を無くすと、forが動く。この文があるとforが動かない。 24 25 i = 0 26 for row in reader: 27 if i == cnt - 1: 28 print("last") 29 else: 30 print(row) 31 i += 1 32
試したこと
cntでreaderの行数を数えたのち、再度withステートメントで開きなおすと、
やりたいことは実現できました。ただ、何度もwith書くのもなぁというのもあり・・・
回答1件
あなたの回答
tips
プレビュー