Python Pandasで特定のテキストを含む行を反復処理して取得する方法

特定のテキストを含む行を反復処理して取得するには、itertuples()メソッドとfind()メソッドを使用します。itertuples()は、DataFrameの行を1行ずつ順番に処理するための便利なメソッドです。

Pandasのインポート

まず、必要なライブラリをエイリアス付きでインポートします。

import pandas as pd

CSVファイルの読み込み

今回は、デスクトップに保存された以下のパスにあるCSVファイルを使用します。

C:\Users\amit_\Desktop\CarRecords.csv

CSVファイルを読み込んで、PandasのDataFrameを作成しましょう。

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

特定のテキストを含む行の取得

次に、DataFrameの行を反復処理して、特定のテキストを含む行を取得します。ここでは、「Car」列に「Lamborghini」というテキストが含まれる行を検索しています。

for k in dataFrame.itertuples():
    if k[1].find('Lamborghini') != -1:
        print(k)

find()メソッドは、指定した文字列が見つかった場合はその位置（インデックス）を返し、見つからなかった場合は-1を返します。そのため、戻り値が-1以外であれば、目的のテキストが含まれていると判断できます。

完全なコード例

以下に、全体のコードを示します。

import pandas as pd

# CSVファイルを読み込む
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\CarRecords.csv")
print("DataFrame...\n", dataFrame)

# 特定のテキストを含む行を反復処理して取得する
# Car列から「Lamborghini」というテキストを検索する
for k in dataFrame.itertuples():
    if k[1].find('Lamborghini') != -1:
        print(k)

実行結果

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

Pandas(Index=5, Car='Lamborghini', Place='Chandigarh', UnitsSold=80)
Pandas(Index=8, Car='Lamborghini', Place='Delhi', UnitsSold=100)

このように、「Lamborghini」を含む行だけが抽出されて表示されていることがわかります。各行はPandasという名前付きタプルとして出力され、インデックスや各列の値を確認できます。