我正在编写一个脚本,该脚本应该将文件作为输入,将其读入pandas数据帧,然后执行一些操作。我本来希望不强制文件采用特定的格式,并让用户告诉脚本它应该如何读取它。但是,我似乎没有找到任何支持。可能吗?或者我应该手动完成吗?谢谢你的帮助。
答案 0 :(得分:1)
也许是这些方面的东西:
filename = "userprovided.csv"
mapping = {
'csv': pd.read_csv,
'xlsx': pd.read_excel,
...
}
ext = os.path.splitext(filename)[1]
data = mapping[ext](filename)
当然需要进行一些错误检查等。