for文で想定通りの結果を得る方法

Question

下記コードをfor文（or シンプルな方法）で記載する方法がありましたら教えてください。
（本件とは関係ないですが参考として、join_dfはdf.mergeとほぼ同じです）

train = join_df(train, weather)
test = join_df(test, weather)

私がやっていた方法は、

for t in (train, test):
    t = join_df(t, weather) 

となりますが、't' tableには変更が反映されるものの、
train, testには変更が反映されません。

複数テーブルをmergeする必要があり、
for文などで簡潔に表せる方法が知りたいためご質問させていただきます。

Answer 1

train, test = [join_df(t, weather) for t in (train, test)]

Answer 2

おそらくやりたいことをそのまま実現するのはPythonでは厳しいです。

train（やtest）変数は内部的には実データへのIDしか保持していません。
そして実データに何らかの演算を施すと、新たに生成された実データ＋それに紐づいた別IDが返されます。
この時、trainは以前のID値を持っているのでtrainの指す実データは演算前のものになります。

ということで、辞書にて名称とデータを紐づける手法をとるのがPython的には妥当です。

def join_df(df_left,df_right,col):
    return df_left.merge(df_right,left_on=col,right_on=col) # 結合

weather = pd.DataFrame({'key':[1,2,3],'a':[11,22,33]})

dfs = { 'train' : pd.DataFrame({'key':[1,2,3],'t':[4,5,6]}),
        'test'  : pd.DataFrame({'key':[1,2,3],'t':[7,8,9]})}

for k,v in dfs.items():
    dfs[k] = join_df(v,weather,'key')

print(dfs['train'])
"""
   key  t   a
0    1  4  11
1    2  5  22
2    3  6  33
"""

print(dfs['test'])
"""
   key  t   a
0    1  7  11
1    2  8  22
2    3  9  33
"""