PandasにおけるDataFrameの最初のN行を取得する

PandasにおけるDataFrameの最初のN行を取得してみます。

■Python

今回のPythonのバージョンは、「3.8.5」を使用しています。（Windows10）（pythonランチャーでの確認）

■DataFrameを作成する

DataFrameの最初のN行を取得する前に、DataFrameを作成します。

■コード

import pandas as pd

data = {
    'Name':['Peach','Apple','Strawberry','Grape','Pear'],
    'Number':[10,20,30,40,50]
}

df = pd.DataFrame(data,columns = ['Name','Number'])

print(df)

インポートでPandasモジュールを呼び出します。dataという変数を作成し、その中に、今回は「Name」と「Number」という2つの行を追加し格納します。

格納後、dfという変数を作成し、pd.DataFrame()と記述し、格納したDataを元にDataFrameを作成。作成後、dfという変数に格納します。

■実行

このスクリプトを実行すると、作成したDataFrameが出力されます。

■DataFrameの最初のN行を取得する

DataFrameの用意ができましたので、DataFrameの最初のN行を取得するスクリプトを書いていきます。

■コード

import pandas as pd

data = {
    'Name':['Peach','Apple','Strawberry','Grape','Pear'],
    'Number':[10,20,30,40,50]
}

df = pd.DataFrame(data,columns = ['Name','Number'])

get_row = df.head(3)

print(get_row)

dfという変数を作成し、pd.DataFrame()と記述し、格納したDataを元にDataFrameを作成。作成後、dfという変数に格納。

その後に、最初のN行を取得するために、get_rowという変数を作成し、その中に「df.head()」と記述します。これで最初のN行を取得することができます。今回は最初の3行を取得してみますので、括弧内に「3」と記述します。

最後に取得した行をprint関数で出力します。

■実行

このスクリプトを「df_n_test_2.py」という名前で保存し、コマンドプロンプトから実行してみます。

実行してみると、DataFrameの最初の3行が取得され出力できることを確認できました。