根据用户输入来命名熊猫DF

Question

我有一个脚本，该脚本从mySQL服务器获取数据并将其另存为pandas数据框，然后必须使用用户选择的唯一名称保存该特定数据框。

数据框是作为“ createDF”函数的一部分创建的，然后保存为“ df”。

df = createDF()

这有效，数据框对象另存为“ df”。我需要做的是允许用户命名数据框，而不用名称“ df”修复它，而是将其另存为“ df + userinput”。其中“ userinput”是用户提供的字符串/整数。

我似乎确定问题出在试图将输入类型（str或int）与对象名称“ df”组合在一起，因为我不断收到“ SyntaxError：无法分配给运算符”的问题。

我不知道可行的解决方案会是什么样，将不胜感激。

Answer 1

我认为维护dictionary数据帧将是一个更好的解决方案。使用密钥userinput将每个数据帧存储在字典对象中。您以后可以轻松地从字典对象访问任何数据框。

dfs = { }
dfs[userinput] = createDF() # add new dataframe
...
print(dfs[userinput]) # access the dataframe