python - 使用 Dictionary (key) 作为 user 的输入，然后从同一个 Dictionary 加载 .csv files(value)

Question

这是它的样子：

dictionary = { '1' : 'mine.csv' , '2' : 'yours.csv' , '3' : 'ours.csv' ...}

我想询问用户输入：

def get_filters():

question = input('which file would you like to view its contents ?' , 1)

所以答案应该是：'1'>>它从字典中打开第一个文件以启动过滤器。

我希望从单个代码打开所有文件：

def load_data(question, ..):
    df= pd.read_csv(dictionary[question]) 

因为 df 将用于过滤文件内的所有数据。

但这不会加载文件，并且程序将过滤输出视为元组。

如何连接字典键和值来加载数据，例如：如果输入是键：加载值。或者其他简单的方法来做到这一点？

Answer 1

我不是 100% 确定我理解这个问题，但你可以使用这样的东西：

def load_data():
    while True:
        question = input('which file would you like to view its contents ?')
        try:
            df = pd.read_csv(dictionary[question])
            return df
        except Exception as e:
            print(e)

这将接受用户输入，尝试打开文件。如果成功，它将返回 df。如果失败，它将打印错误，然后让用户重试。