pandasデータフレームの結合に関して、for文で結合の処理を回したいです。

Question

前提・実現したいこと

pandasデータフレームの結合に関して、for文で結合の処理を回したいのですが、
うまくいかない為、ご教示お願いいたします。

該当ｺｰﾄﾞ

for df,main_file in zip([dfA,dfA1,dfA2],[mainA,mainB,mainC]):
           
            results = df.merge(main_file, how = "left", left_on = "id", right_on ="id")
           #以下削除などの処理があります
            for result in results:
                 return result

こうしましたが、３つ目のデータの先頭行しかとれませんでした。
reusultsで取り出すと、３つめのデータだけとれましたが、他の２つのデータをとれませんでした。
やりたいこととしては、３つのデータをとりだして、それぞれ、エクセルファイルで抽出したいです。
エクセル名は、for 文のmain_fileと同じにしたいです。

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

エラーメッセージ

該当のソースコード

ソースコード

Answer 1

returnすると、その時点で関数内の処理は終了します。
途中で抜けたくなければ、returnはループを抜けてからやりましょう。

ループの部分のコードは以下のようにすればよくなるでしょう。

results = []
for df,main_file in zip([dfA,dfA1,dfA2],[mainA,mainB,mainC]):
    result = df.merge(main_file, how = "left", left_on = "id", right_on ="id")
           #以下削除などの処理があります
    results.append(result)
return results

• それぞれ、エクセルファイルで抽出したいです。
  エクセル名は、for 文のmain_fileと同じにしたいです。

これはまずやってみてください。
うまくいかないなら別の質問を立ててください。

Comments

[oobutamaru]

出来ました。
回答ありがとうございました。