python - break を使用してリストの最初の行を読み取る

Question

ファイルの最初の行だけを読んで何かをしたいとしましょう。私は次のように言います。

stuff = open('some_file.txt','rb')

for row in stuff:
    # do something to first line
    break
for row in stuff:
    # do something to all other lines

そして、これはメモリ内の何かではなくファイルオブジェクトを読んでいるのと同じくらい孤独に機能します。

これを行うことに反対する議論はありますか？

Answer 1

next()次の関数で最初の行を取得できます。

firstline = next(stuff, '')  # default value '' if the file is empty

for row in stuff:
    # rest of the file

これは、あらゆるイテラブルで機能します。行のリストがある場合はlines[0]、 、または最初にリストを呼び出すことができますiter()。

Answer 2

fileinputは優れたモジュールです。

for line in fileinput.input(['some_file.txt']):
  if not fileinput.isfirstline():
    data = proces_line(line);
    output(data)