【Python Pandas】pdpipeでパイプラインを作成し、DataFrameから列を削除する方法

PandasのDataFrameから特定の列（カラム）を削除したい場合、pdpipeライブラリのColDrop()メソッドを利用すると便利です。pdpipeはデータ処理の各ステップをパイプラインとして管理できるライブラリで、列の削除や行の削除といった操作を簡潔に記述できます。

この記事では、pdpipeを使ってパイプラインを作成し、DataFrameから列を削除する手順をサンプルコードとともに解説します。

必要なライブラリのインポート

まず、pdpipeとpandasをそれぞれ別名（エイリアス）をつけてインポートします。

import pdpipe as pdp
import pandas as pd

サンプル用のDataFrameを作成する

次に、「Car」と「Units」の2つの列を持つDataFrameを作成しましょう。

dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
        "Units": [100, 150, 110, 80, 110, 90]
    }
)

ColDrop()メソッドで列を削除する

DataFrameから列を削除するには、pdp.ColDrop()メソッドを使用します。引数に削除したい列名を指定し、apply()メソッドで対象のDataFrameに適用します。

ここでは「Units」列を削除してみます。

resDF = pdp.ColDrop("Units").apply(dataFrame)

完全なサンプルコード

以下は、行の削除（ValDrop）と列の削除（ColDrop）を組み合わせた完全なコード例です。

import pdpipe as pdp
import pandas as pd

# DataFrameを作成
dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
        "Units": [100, 150, 110, 80, 110, 90]
    }
)

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

# pdp.ValDrop()で特定の値を持つ行を削除
dataFrame = pdp.ValDrop(['Jaguar'], 'Car').apply(dataFrame)
print("\n行を削除した後のDataFrame...\n", dataFrame)

# pdp.ColDrop()で列を削除
resDF = pdp.ColDrop("Units").apply(dataFrame)
print("\n列を削除した後のDataFrame...\n", resDF)

実行結果

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

DataFrame ...
        Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

行を削除した後のDataFrame...
        Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110

列を削除した後のDataFrame...
        Car
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley

まとめ

pdpipeのColDrop()メソッドを使えば、DataFrameから指定した列を簡単に削除できます。同様にValDrop()を使えば、特定の値を持つ行を削除することも可能です。これらの操作をパイプラインとして組み合わせることで、複数の前処理ステップを整理された形で記述でき、コードの可読性と再利用性が向上します。データ分析の前処理を効率化したい方は、ぜひ活用してみてください。