Dive into hoge

データ分析関連の備忘録

pandas操作⑧:queryに変数を指定する場合

queryで条件指定にリストを使いたくて、生のリストではなく変数を入れたい場合。

# tmpはリスト
何らかのデータフレーム.query('カラム名==tmp')

そのまま変数入れるとエラーになる。

UndefinedVariableError: name 'tmp' is not defined

下記のように変数の前に@を入れて回避できる。

# tmpはリスト
何らかのデータフレーム.query('カラム名==@tmp')

<参考>
nekoyukimmm.hatenablog.com