自然言語処理100本ノック2020から逃げるな「14. 先頭からN行を出力」

これは自然言語処理100本ノック2020に挑んだ、とある人間の記録
他の問題はこちら → 自然言語処理100本ノック2020から逃げるな まとめ

14. 先頭からN行を出力

自然数Nをコマンドライン引数などの手段で受け取り,入力のうち先頭のN行だけを表示せよ.確認にはheadコマンドを用いよ.

Pythonコード

import sys

path = sys.argv[1]
N = int(input('N = '))

with open(path) as f:
    for i in range(N):
        print(f.readline(), end='')

確認用UNIXコマンド

私の環境は少し特殊(fish)なので、もしかしたら動かないかも。

head -n $1 ../popular-names.txt

$1はNの値。

実行結果

N = 5
Mary    F       7065    1880
Anna    F       2604    1880
Emma    F       2003    1880
Elizabeth       F       1939    1880
Minnie  F       1746    1880