【pandas】リスト内の全てのデータでデータフレームへフィルターをかけてデータを抽出し、Excelへそれぞれ別シートで張り付ける方法

Question

前提・実現したいこと

【pandas】リスト内の全てのデータでデータフレームへフィルターをかけてデータを抽出し、Excelへそれぞれ別シートで張り付ける方法

発生している問題・エラーメッセージ

name 'i' is not defined

該当のソースコード

　　A　　　B　
0 リンゴ　東京
1 ナス　　東京
2 バナナ　大阪
3 リンゴ　奈良
4 ナス　　埼玉
5 イチゴ　大阪

というデータフレームdfについて、
Aの重複をuniqueで削除してリスト化します。

food_list = ['リンゴ'、'ナス'、'バナナ'、'イチゴ']

その後、
リンゴであればAにリンゴとある行のみをフィルターして、df_ringoを作成し、
エクセルへシート名 = リンゴでデータを貼り付け、
次にナスでフィルターして同様にdf_nasuを作成して、
リンゴと同じブック内に別シートでシート名 = ナスで貼り付け・・・繰り返しして、
リスト内の4種類でフィルターしたデータを1つのブック内に別シートで貼り付けたいと思っています。

for i in range(4):
    df_i = df.query('A == @food_list[i]')
    with pd.ExcelWriter('データ/foods.xlsx') as writer:
        df_i.to_excel(writer, sheet_name = food_list[i])

上記のようなfor構文を作成したのですが、
name 'i' is not defined

というエラーが出てしまっております。
以上、1点アドバイスよろしくお願い致します。		

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[meg_]

「name 'i' is not defined」のエラーが発生したのはコードのどの行ですか？

[退会済みユーザー]

meg_様
ご回答ありがとうございます。わかりにくくて申し訳ございません。
df_i = df.query('A == @food_list[i]')
の部分になりますが、@iとすることでエラーなくなりました。

Answer 1

df_i = df.query('A == @food_list[i]')

を

df_i = df.query('A == @food_list[@i]')

に変更しましょう。

food_list[i] という式は、food_listという変数に入っているlistからiという変数に入っているintをindexとする要素を取り出すという意味なので、queryで使う場合には、どちらにも@が必要です。

Comments

[退会済みユーザー]

ppaul様　ありがとうございます。
iにも@をつけるのは盲点でした。

for i in range(3):
d_i = df.query('A == @food_list[@i]')
with pd.ExcelWriter('データ/foods.xlsx') as writer:
d_i.to_excel(writer, sheet_name = food_list[i])
とした場合、i=2のシートしかエクセルファイル内に生成されない
(i=0,i=1のシートへ上書きされている？)
のですが、with以降の関数の書き方が間違っているのでしょうか....??

[ppaul]

以降と言うより，順番です。forの中でファイルを開いているので上書きされます。
with pd.ExcelWriter('データ/foods.xlsx') as writer:
for i in range(4):
df_i = df.query('A == @food_list[i]')
df_i.to_excel(writer, sheet_name = food_list[i])
と書くべきですね。

[退会済みユーザー]

ppaul様
順番は全く意識していなかったです...
上記のコードへ修正したら上手くシート生成できました。
ご教授いただきまして誠にありがとうございましたm(__)m