Pandasで条件を指定して行を取り出す方法

Pandasで条件を指定して行を取り出す方法を紹介します。

使用するデータフレーム

data変数に以下のデータフレームが入っていると仮定します。

	Name	　Age	Gender	Address
0	田中　太郎	　17	　　　male	東京
1	鈴木　一郎	　43	　　　male	大阪
2	渋井丸　巧	　20	　　　male	渋谷
3	社畜　　丸	　22	　　　male	会社
4	宗竜　レイ	　28	　　　female	日本

書き方

data[data["条件にしたいカラム名"] 条件式]

条件を指定して行を取り出す

では、実際に条件を指定して行を取り出してみましょう。
dataのAgeが21歳以上の行を取り出してみます。

data[data["Age"] >= 21]

	Name	　Age	Gender	Address
1	鈴木　一郎	　43	　　　male	大阪
3	社畜　　丸	　22	　　　male	会社
4	宗竜　レイ	　28	　　　female	日本

無事Ageが21以上の行だけ取り出せましたね。

一度に複数条件を指定して行を取り出す方法

条件ごとに()で括り、&もしくは|で繋げます。
Ageが21以上で、Genderがmaleの行を取り出してみます。

data[(data["Age"] >= 21) & (data["Gender"] == "male")]

	Name	　Age	Gender	Address
1	鈴木　一郎	　43	　　　male	大阪
3	社畜　　丸	　22	　　　male	会社

()で括らないとエラーが出るので注意してください。