Python Pandas – iloc()でインデックスに基づいて特定の値を選択しサブセットを作成する方法

Pandasでインデックス(位置番号)に基づいて列から特定の値を選択し、サブセット(DataFrameの一部)を作成するには、iloc()メソッドを使用します。ilocは「integer location」の略で、行番号や列番号といった整数位置を指定してデータを抽出できる便利なメソッドです。

まず、pandasライブラリをインポートします。

import pandas as pd

次に、商品レコードを含むPandas DataFrameを作成します。このDataFrameには「Product(商品名)」「Opening_Stock(期首在庫)」「Closing_Stock(期末在庫)」の3つの列があります。

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],
                          "Opening_Stock": [300, 700, 1200, 1500],
                          "Closing_Stock": [200, 500, 1000, 900]})

iloc()でサブセットを作成する

iloc()を使えば、「先頭2行 × 最初の2列」といった範囲を簡単に指定できます。行と列の位置をスライス形式で渡すのがポイントです。

print("\niloc()で表示したサブセット = \n", dataFrame.iloc[0:2, 0:2])

完全なサンプルコード

以下が一連の処理をまとめた完全なコード例です。

import pandas as pd

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],
                          "Opening_Stock": [300, 700, 1200, 1500],
                          "Closing_Stock": [200, 500, 1000, 900]})

print("DataFrame...\n", dataFrame)

print("\nProduct列のみを表示:\n", dataFrame['Product'])

# iloc()を使って2列 × 先頭2行のサブセットを作成
print("\niloc()で表示したサブセット = \n", dataFrame.iloc[0:2, 0:2])

実行結果

上記のコードを実行すると、次のような出力が得られます。

DataFrame...
    Closing_Stock  Opening_Stock     Product
0            200            300    SmartTV
1            500            700  ChromeCast
2           1000           1200     Speaker
3            900           1500   Earphone

Product列のみを表示:
0        SmartTV
1    ChromeCast
2       Speaker
3      Earphone
Name: Product, dtype: object

iloc()で表示したサブセット =
   Closing_Stock  Opening_Stock
0            200            300
1            500            700

まとめ

dataFrame.iloc[0:2, 0:2]のように指定すると、0行目から1行目まで(2行目の手前まで)かつ0列目から1列目まで(2列目の手前まで)の範囲が抽出されます。このようにiloc()を活用すれば、行番号と列番号を自由に組み合わせて、目的のデータ部分だけを柔軟に取り出すことができます。