【Python】言語処理100本ノック2020 14.先頭からN行を出力

問題

nlp100.github.io

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

環境

Macbook
Python3.7.7

ソースコード

GitHubを載せておきます。

github.com

あくまで一例です。
もっと効率的な書き方は調べると出てきます。 （ただ個人的には難しかった…）

upura.hatenablog.com

import pandas as pd

#tableの作成
df = pd.read_table('chart2/popular-names.txt',header=None,names=['name','sex','number','year'])


def output_head(N):
    print(df.head(N))

N = input("数字を入力してください>>")
output_head(int(N))

pandasでtableを前の通り読み込みます。
このpandasにある、head()メソッド を使えば、引数で渡された数だけ、
先頭からデータを読み込みます。

おわり。
pandasめちゃくちゃ便利…というお話。

note.nkmk.me