ファイルの最初の N 行を取得する

大きな生データ ファイルを扱う場合、多くの場合、特定の数値を抽出する必要があります。さらに処理または分析するための行数。 Python では、このタスクを達成するための複数のアプローチがあります。

リスト内包表記を使用した最初の N 行の読み取り

シンプルで効果的な方法には、リスト内包表記の利用が含まれます。

with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]
print(head)

このアプローチでは、next() 関数を使用して入力ファイルを反復処理し、最初の lines_number 行を head リストに保存します。

islice() 関数の使用

Python の itertools モジュールを利用する別のアプローチ:

from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))
print(head)

ここでは、 islice() 関数を使用して入力ファイルの最初の names_number 行を反復処理し、抽出された行のリストを作成します。

効果オペレーティング システム

上記の実装は、使用されているオペレーティング システムに関係なく機能します。ただし、Python 2 では next() 関数は xrange() として知られており、古いコード ベースでは対応する調整が必要になる場合があることに注意してください。