从用户输入调用特定的 Pandas Dataframe 以在函数中使用？

Question

这可能已经得到解答，但我无法完全找到合适的单词组来搜索以找到我遇到的问题的答案。

情况： 我有几个可以插入到函数中的数据框。该函数要求我命名数据框，以便它能够成形。

def heatmap(table, cmap=cm.inferno, vmin=None, vmax=None, inner_r=0.25, pie_args={}:
     n, m = table.shape

我希望用户能够指定用作表格的数据框，如下所示：

table_name= input('specify Table to Graph: ')
heatmap(table_name)

期望：如果用户输入是 TableXYZ，则变量 table_name 将引用 TableXYZ，以便该函数能够找到TableXYZ 的形状以在函数的其他部分使用该信息。

实际发生的情况：当我尝试运行代码时，我收到“AttribureError：'str'没有属性'shape'”。我看到 table_name 输入是一个字符串对象，但我试图引用数据框本身，而不是名称。

我觉得我缺少将用户的输入转变为函数实际上可以呈现的形式的步骤。

Answer 1

我建议将您的每个数据范围分配给字典，然后通过名称从字典中检索数据框，以将其传递到heatmap函数。

例如：

df_by_name = {"df_a": pd.DataFrame(), "df_b": pd.DataFrame()}
table_name= input('specify Table to Graph: ')
df = df_by_name[table_name]
heatmap(df)

在第一行中替换pd.dataframe（），使用您要选择的实际数据框架。