Pandas DataFrameの行のグループにアクセスする方法｜loc()の使い方を解説

PandasのDataFrameで特定の行のグループにアクセスするには、loc()メソッドを使用します。例えばdf.loc[2:5]と指定すると、ラベル2から5までのすべての行を選択できます。

手順

• 二次元・サイズ可変・異なるデータ型を格納できる表形式データとして、DataFrame dfを作成します。
• 入力となるDataFrame dfを出力して確認します。
• df.loc[2:5]を使って、ラベル2〜5の行を選択します。
• 選択後のDataFrameを出力します。

サンプルコード

import pandas as pd

df = pd.DataFrame(
   {
      "x": [5, 2, 7, 0, 7, 0, 5, 2],
      "y": [4, 7, 5, 1, 5, 1, 4, 7],
      "z": [9, 3, 5, 1, 5, 1, 9, 3]
   }
)

print("入力DataFrame:\n", df)
df = df.loc[2:5]
print("新しいDataFrame:\n", df)

実行結果

入力DataFrame:
   x  y  z
0  5  4  9
1  2  7  3
2  7  5  5
3  0  1  1
4  7  5  5
5  0  1  1
6  5  4  9
7  2  7  3

新しいDataFrame:
   x  y  z
2  7  5  5
3  0  1  1
4  7  5  5
5  0  1  1

補足：locとilocの違い

locはラベルベースのインデックス指定であり、範囲指定した場合は終端のラベルも含まれます。一方、ilocは位置（行番号）ベースの指定で、Pythonのスライスと同様に終端は含まれません。用途に応じて両者を使い分けることで、意図した通りの行抽出が可能になります。