言語処理100本ノック 2020「15. 末尾のN行を出力」

問題文

nlp100.github.io

問題の概要

「14. 先頭からN行を出力」とほぼ同様です。今回表示するのは上位N件ではなく下位N件なので「head(N)」の代わりに「tail(N)」を使います。

import sys
import pandas as pd


if len(sys.argv) == 1:
    print('Set arg n, like "python ch02/ans15.py 5"')
else:
    n = int(sys.argv[1])
    df = pd.read_csv('ch02/popular-names.txt', sep='\t', header=None)
    print(df.tail(n))