例題データは、機械学習でお馴染みのアイリスデータセットを用いました。150行×5列のデータです。
本記事でやりたいことは、例えば、sepal_lengthという列（カラム）に対して、5, 5.5, 6の値である行を全て抽出したい場合に適用し、次のようにします。リストの直前に@マークを付けることでリストと認識することがポイントです。
target_list = [5, 5.5, 6]
df2 = df.query('sepal_length in @target_list')
print(df2)
下図はその結果です。sepal_length列が5, 5.5, 6である行が全て一括で取得できることがわかります。
以上
<広告>
リンク
