csvの1行目と最終行のみをprintしたいです。

Question

前提・実現したいこと

csvの1行目と最終行をprintしたいです。

発生している問題・エラーメッセージ

1行目のみ、最終行（181行目）のみをprintすると問題なく表示されますが、
どちらも表示させようとすると以下のエラーが出ます。

lastrow = csvfile.readlines()[180]
IndexError: list index out of range

該当のソースコード

import csv
csvfile = open("01_01.csv")
firstrow = csvfile.readlines()[0]
lastrow = csvfile.readlines()[180]
print(firstrow)
print(lastrow)

試したこと

以下のどちらかのみを残して実行すると正しく表示されました。

firstrow = csvfile.readlines()[0]
print(firstrow)

lastrow = csvfile.readlines()[180]
print(lastrow)

Answer 1

firstrow = csvfile.readlines()[0]
lastrow = csvfile.readlines()[180]

一度目のreadlinesでファイルの最後まで読み切ってしまっているので、2度目のreadlinesは意図したようになりません。readlinesした結果を最初に温存しておかなければだめです。

data = csvfile.readlines()
print(data[0])
print(data[180])

Comments

[matya]

早速のご回答ありがとうございます！
やってみたところ、できました！

Answer 2

質問文にあるコードでは、csv ファイル全体がメモリーの読み込まれます。
問文のような動作をさえるだけなら、メモリーには一行だけをもつだけで十分です。

一行文だけを読む処理を ループさせ、最初の行とさいごの行のときだけ print するというコードでたいおうができると思います。

web 上で python csv fuge などで検索すると巨大な csv を扱う工夫をいろいろ見つけることができます。

Comments

[matya]

わかりました！
ありがとうございます！

Answer 3

fileオブジェクトは内部でどこまで読み込んだかという情報（ファイルポインタ）を持っています。
よってreadlines()を２回呼び出した場合、１回目は最初から最後まで、２回目は最後から読み込むので結果は空になります。
意図した動作をしたい場合は以下のようにすると良いでしょう。

import csv
csvfile = open("01_01.csv")

rows = csvfile.readlines()
print(rows) # ['あいう\n','かきく\n,....]

print(csvfile.readlines()) # [] 空となる！

print(rows[0]) # 先頭
print(rows[-1])# 最後

csvfile.close()

Comments

[matya]

ご回答ありがとうございます！
そういうことだったんですね…

解決しました！