以编程方式设置要使用pandas读取的文件类型？

Question

我正在编写一个脚本，该脚本应该将文件作为输入，将其读入pandas数据帧，然后执行一些操作。我本来希望不强制文件采用特定的格式，并让用户告诉脚本它应该如何读取它。但是，我似乎没有找到任何支持。可能吗？或者我应该手动完成吗？谢谢你的帮助。

Answer 1

也许是这些方面的东西：

filename = "userprovided.csv"

mapping = {
    'csv': pd.read_csv,
    'xlsx': pd.read_excel,
    ...
}

ext = os.path.splitext(filename)[1]

data = mapping[ext](filename)

当然需要进行一些错误检查等。