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

时间:2017-03-14 12:04:55

标签: python pandas dataframe

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

1 个答案:

答案 0 :(得分:1)

也许是这些方面的东西:

filename = "userprovided.csv"

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

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

data = mapping[ext](filename)

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