【python】csvモジュールのreaderにてforが動かない

Question

前提・実現したいこと

python学習中です（初投稿につき無礼がありましたら申し訳ありません）。
csvモジュールをいじっているのですが、readerの挙動で不明な点があります。

forを使って一行ずつ読み取っていき、
指定の行数に至ったときに別処理をする、ということを実現したいのですが、
readerとfor文の間にreaderを扱うコードを記載すると、
以降のfor文が動きません（エラーも出ずに、何の処理もせず終了する）。

調べてみたのですが原因が分からず、ご教示いただけると幸いです。

該当のソースコード

import csv
import os

if os.path.exists("./csv_train.csv"):
    pass

else:
    with open("csv_train.csv", "w", newline="") as csv_file:
        fieldnames = ["NAME", "COUNT"]
        writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
        writer.writeheader()

        writer.writerow({"NAME": "aaa", "COUNT": 1})
        writer.writerow({"NAME": "aaa", "COUNT": 1})
        writer.writerow({"NAME": "aaa", "COUNT": 1})

with open("csv_train.csv", "r") as csv_file:
    reader = csv.DictReader(csv_file)

    #↓を無くすと、forが動く。この文があるとforが動かない。
    cnt = len(list(reader))
　　#↑を無くすと、forが動く。この文があるとforが動かない。

    i = 0
    for row in reader:
        if i == cnt - 1:
            print("last")
        else:
            print(row)
        i += 1

試したこと

cntでreaderの行数を数えたのち、再度withステートメントで開きなおすと、
やりたいことは実現できました。ただ、何度もwith書くのもなぁというのもあり・・・

Comments

[takasima20]

pandas 使ったら?

[cube107]

ありがとうございます。pandasも勉強せねばと思っているところなのですが、不思議な動きだなと思ったので質問させていただきました。

Answer 1

イテレータは状態を記憶するので、消費すると再度使えないのです。

>>> it = iter(range(3))
>>> list(it)
[0, 1, 2]
>>> list(it)
[]

簡単に解決するのなら、リストにして使ってしまえば良いです。ただしメモリは圧迫します。

with open("csv_train.csv", "r") as csv_file:
    reader = csv.DictReader(csv_file)

    lines = list(reader)
    cnt = len(lines)

    i = 0
    for row in lines:
        if i == cnt - 1:
            print("last")
        else:
            print(row)
        i += 1

Comments

[cube107]

お早いご回答ありがとうございます！
イテレータはリストと違い「今何番目を選んでいるのか」の状態を保持するもの、というようなイメージでしょうか？
readerは辞書型だと勝手に思い込んでいたのですが、printしても<csv.DictReader object at 0x0000013FA3AA4C88>等、表示されるのはイテレータだったからなのですね…！勉強になります。ありがとうございます。