根据用户输入选择合适的数据框

Question

我加载了以下数据帧：

1. df_1000-2000
2. df_3000-4000
3. df_5000-6000
4. df_7000-8000

现在我得到的用户输入值为1000-2000。基于用户输入值，我需要处理相应的数据帧。 在这种情况下，我需要处理：df_1000-2000

如何根据用户输入动态选择数据框并开始处理？

Answer 1

使用字典

您应该重组存储和访问数据框的方式。首先定义一个字典：

dfs = {'1000-2000': df_1000-2000, '3000-4000': df_3000-4000, etc.}

然后接受用户输入并使用它来查询字典很简单：

value = input('Input the range you require, e.g. 1000-2000:')
res = dfs[value]