Python Pandas入門：インデックス演算子（角括弧）でサブセットのDataFrameを作成する方法

PandasでサブセットのDataFrame（データフレームの一部）を作成するには、インデックス演算子である角括弧「[ ]」を使用します。まずは、3つの列を持つPandasのDataFrameを作成してみましょう。

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

このDataFrameには、「Product（製品名）」「Opening_Stock（期首在庫）」「Closing_Stock（期末在庫）」の3つの列が含まれています。

単一の列でサブセットを作成する

1つの列だけを抽出してサブセットを作成する場合は、次のように列名を二重の角括弧で囲みます。

dataFrame[['Product']]

複数の列でサブセットを作成する

複数の列を抽出したい場合は、列名をリスト形式で指定します。

dataFrame[['Opening_Stock','Closing_Stock']]

なお、角括弧を1重にした場合（例：dataFrame['Product']）はSeriesが返されますが、二重の角括弧 [[ ]] を使うとDataFrameとして取得できる点に注意してください。

サンプルコード（完全版）

以下に、ここまでの内容をまとめた完全なコードを示します。

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("\nDisplaying a subset using indexing operator:\n",dataFrame[['Product']])

print("\nDisplaying a subset with multiple columns:\n",dataFrame[['Opening_Stock','Closing_Stock']])

実行結果

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

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

Displaying a subset using indexing operator:
        Product
0      SmartTV
1   ChromeCast
2       Speaker
3     Earphone

Displaying a subset with multiple columns:
    Opening_Stock   Closing_Stock
0              300             200
1              700             500
2             1200            1000
3             1500             900