Python Pandas - 合計値に基づいてDataFrameから行をフィルタリングする方法

はじめに

PandasのDataFrameでは、行ごとの合計値を条件としてフィルタリングを行うことができます。本記事では、3人の生徒の点数データを例に、「各行の合計点が200を超える」行だけを抽出する方法を解説します。逆に、合計が200未満の行を除外したい場合にも同じ手法が使えます。

DataFrameの作成

まず、Jacob・Ted・Jamieの3人の生徒の点数を格納した、3列構成のDataFrameを作成します。

import pandas as pd

# 3列のDataFrameを作成
dataFrame = pd.DataFrame({
    'Jacob_Marks': [95, 90, 70, 85, 88],
    'Ted_Marks': [60, 50, 65, 85, 70],
    'Jamie_Marks': [77, 76, 60, 45, 50]
})

print("元のDataFrame:\n", dataFrame)

合計値に基づくフィルタリングの方法

行方向の合計は sum(axis=1) で計算できます。この値と比較条件を組み合わせたブールマスクをDataFrameに渡すことで、条件を満たす行だけを簡単に抽出できます。

# 各行の合計が200を超える行のみを取得
dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]

完全なコード例

以下に、作成からフィルタリングまでの一連の処理をまとめた完全なコードを示します。

import pandas as pd

# 3列のDataFrameを作成
dataFrame = pd.DataFrame({
    'Jacob_Marks': [95, 90, 70, 85, 88],
    'Ted_Marks': [60, 50, 65, 85, 70],
    'Jamie_Marks': [77, 76, 60, 45, 50]
})

print("元のDataFrame:\n", dataFrame)

# 行ごとの合計が200を超える行のみを抽出
dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]

print("更新後のDataFrame:\n", dataFrame)

実行結果

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

元のDataFrame...
    Jacob_Marks  Jamie_Marks  Ted_Marks
0           95           77         60
1           90           76         50
2           70           60         65
3           85           45         85
4           88           50         70

更新後のDataFrame...
    Jacob_Marks  Jamie_Marks  Ted_Marks
0           95           77         60
1           90           76         50
3           85           45         85
4           88           50         70

出力を確認すると、インデックス2の行（合計195点）が除外され、合計が200を超える4つの行だけが残っていることがわかります。このように sum(axis=1) とブールインデックスを組み合わせることで、行単位の合計値に基づいた柔軟なデータ抽出が可能になります。