Pandas.queryに変数を入れる方法

Question

query関数でデータ任意のデータを抽出したいのですが、変数を入れるとエラーになってしまいます。
どうしたらいいでしょうか?

i = 20010101
df1 = df_sorted[u].query('C_PRICE < A_PRICE & DATE == i')

出ているエラー

UndefinedVariableError: name 'i' is not defined

Answer 1

クエリ文字列中の変数名の前に@マークを付けることで、呼び出し元で定義された変数にアクセスできます。

expr : string
The query string to evaluate. You can refer to variables in the environment by prefixing them with an ‘@’ character like @a + b.
pandas.DataFrame.query — pandas 0.23.4 documentation

質問文のコードはこれで行けるでしょうか。

df1 = df_sorted[u].query('C_PRICE < A_PRICE & DATE == @i')

Answer 2

df1 = df_sorted[u].query('C_PRICE < A_PRICE & DATE == i')の中のiは単なる文字列として解釈されます。
df1 = df_sorted[u].query('C_PRICE < A_PRICE & DATE == {}'.format(i))として文字列にiの値が入るようにしてください。