Python・PandasのDataFrameから複数の列を選択する方法

Pandasでは、CSVファイルから読み込んだDataFrameに対して、二重の角括弧 [[ ]] を使うことで、複数の列を簡単に選択できます。本記事では、具体的な手順とサンプルコードを交えて解説します。

前提：サンプルデータについて

ここでは、Microsoft Excelで開いた以下のような内容のCSVファイル（SalesData.csv）を例にします。このデータには「Car（車名）」「Reg_Price（登録価格）」「Units（台数）」の3つの列が含まれています。

ステップ1：CSVファイルをPandas DataFrameに読み込む

まず、read_csv() メソッドを使って、CSVファイルのデータをPandas DataFrameに読み込みます。

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")

ステップ2：複数の列を選択する

複数の列を選択するには、角括弧を使用します。括弧内に取得したい列名をリスト形式で指定することで、データセット全体から該当する複数の列を抽出できます。

dataFrame[['Reg_Price','Units']]

このように、列名を二重の角括弧で囲むのがポイントです。単一の列の場合は dataFrame['Reg_Price'] のように一重の括弧でも動作しますが、複数列を指定する場合は必ずリストとして渡す必要があります。

完全なサンプルコード

以下は、CSVファイルを読み込んで2つの列（「Reg_Price」と「Units」）を表示するコード全体です。

import pandas as pd

# CSVファイルからPandas DataFrameへデータを読み込む
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nReading the CSV file...\n", dataFrame)

# 2つの列を表示
res = dataFrame[['Reg_Price', 'Units']]
print("\nDisplaying two columns : \n", res)

実行結果

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

Reading the CSV file...
        Car   Reg_Price   Units
0       BMW        2500     100
1     Lexus        3500      80
2      Audi        2500     120
3    Jaguar        2000      70
4   Mustang        2500     110

Displaying two columns :
    Reg_Price   Units
0        2500     100
1        3500      80
2        2500     120
3        2000      70
4        2500     110

出力を見ると、「Car」列が除外され、「Reg_Price」と「Units」の2つの列だけが抽出されていることが確認できます。

まとめ

• Pandas DataFrameから複数の列を選択するには、df[['列名1', '列名2']] のように二重の角括弧を使います。
• 指定した順序で列が返されるため、列の並べ替えにも活用できます。
• 3つ以上の列も同様に、リストに列名を追加するだけで選択可能です。