pandasのDataFrameを用いたときにDataFrame名を取得したい

Question

解決したいこと

pandasのDataFrameを用いたときにDataFrame名を取得したい。下の例でいうとdf_testという名前を取り出したい。今のままだとdf_testのDataFrameの表が返ってくるためうまく表示できない。

##コード例

df_test = pd.DataFrame({ 'name' : (["A","B"]),'score' : 0,})
print("df名は"+game(df_test))


def game(df_1):
    print(df_1)

#実際にはここの関数内に別の関数にdf_testという文字列を引数として渡したい

    return df_1

Comments

[8524ba23]

なぜ別の関数にdf_test変数ではなく「df_test」という文字列を渡したいのでしょうか？

[launcha]

def win_lose(df_test,win1):
for i in range(6):
if df_comic["チーム名"][i]==df_test:
df_comic["試合数"][i]+=1
df_comic["勝数"][i]+=win1

簡略していますが、１チームをdfで表現しているため、別のdfのチーム名と照らし合わせる時に文字列を参照しようと考えていました。

[8524ba23]

よくわかりませんが、チーム名とデータフレームが１対１で紐づけられているようなイメージでしょうか？

[launcha]

そうです

Answer 1

とりあえずこの関数で変数名を返せます。

from inspect import currentframe

def game(*args):
    names = {id(v):k for k,v in currentframe().f_back.f_locals.items()}
    df_ret = names.get(id(args[0]),'???')
    return df_ret

Comments

[launcha]

回答ありがとうございます。この場合、df_testを返すのではなく、関数内で使っているdf_1を返すようになっていますか？

[yureighost]

関数gameに渡した変数の名前が返されるようになっています。
df_1の名前を返したいのであれば、別に関数を定義してgame関数内から呼ぶことになるかと。

[launcha]

なるほど。できました。dataframeで渡せた上で名前も取れるんですね。知りませんでした。ありがとうございます！！！