Pandasで条件で絞り込み必要な行だけ抽出したいことってありますよね。そんなときのやり方をご紹介
df[df[”カラム名"] 比較演算子 条件]で絞り込む
実際に具体例で絞り込みましょう
Titanicで生存者だけを絞り込みたい
- カラム名:”Survived”
- 値は:0 or 1 が入っています 1が生存
Survivedの1の行を抽出したい
絞り込む前と絞り込んだあとの最初の5つを取得します
.head()を後ろにつけます
絞り込む前
df.head()
絞り込んだあとはSurvivedの列が1のみとなっています
ポイントは df[列名]をさらにdf[]で囲って条件指定しているところですね
df[df["Survived"] == 1].head()
条件は pythonの比較演算子と同様に絞り込みができます
以上以下、より大きい、未満、等号、不等号