自然言語処理100本ノック2020から逃げるな「15. 末尾のN行を出力」

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

15. 末尾のN行を出力

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

Pythonコード

import sys

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

with open(path) as f:
    lines = f.readlines()
    print(''.join(lines[-N:]), end='')

確認用UNIXコマンド

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

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

$1はNの値。

実行結果

N = 5
Benjamin        M       13381   2018
Elijah  M       12886   2018
Lucas   M       12585   2018
Mason   M       12435   2018
Logan   M       12352   2018