python : queryをforループ内で回したときのnot definedエラー解消方法

Question

作成したdfの中から条件に合う行が何行あるかを計測するべく下記のコードを書きました。
しかし最後のforループ内のquery条件でiが認識されず下記のエラーが発生してしました。
name 'i' is not defined
正常に動作させるための変更点、又は同じ事が出来るコードをお教えいただけると幸いです。
よろしくお願いいたします。

test = pd.DataFrame(
    data = {
        "one" :[1, 2, 3],
        "two" :[1, 2, 3],
        "three": [1, 2, 3]
    }
)

combination = [1, 2, 3]
count = {}

for i in combination:
    data = test.query("one == i").count()
    count[i] = data["one"]

Answer 1

変数名の前に@を付けます。

pandas.DataFrame.query

You can refer to variables in the environment by prefixing them with an ‘@’ character like @a + b.

test = pd.DataFrame(
    data = {
        "one" :[1, 2, 3],
        "two" :[1, 2, 3],
        "three": [1, 2, 3]
    }
)

combination = [1, 2, 3]
count = {}

for i in combination:
    data = test.query("one == @i").count()
    count[i] = data["one"]

print(count)
# {1: 1, 2: 1, 3: 1}

Comments

[Trader.a]

回答ありがとうございます！
仕様が記載されているページまで教えていただけて大変助かりました。
やはり使い始める前には仕様を勉強する必要がありますね。