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

Python 以编程方式设置要使用pandas读取的文件类型?,python,pandas,dataframe,Python,Pandas,Dataframe,我正在写一个脚本,它应该把一个文件作为输入,读入一个数据帧,然后做一些事情。我不希望强制文件采用特定的格式,而是让用户告诉脚本应该如何读取它。然而,我似乎找不到对此的任何支持。可能吗?还是我应该手动操作?谢谢你的帮助。也许可以这样做: filename = "userprovided.csv" mapping = { 'csv': pd.read_csv, 'xlsx': pd.read_excel, ... } ext = os.path.splitext(file

我正在写一个脚本,它应该把一个文件作为输入,读入一个数据帧,然后做一些事情。我不希望强制文件采用特定的格式,而是让用户告诉脚本应该如何读取它。然而,我似乎找不到对此的任何支持。可能吗?还是我应该手动操作?谢谢你的帮助。

也许可以这样做:

filename = "userprovided.csv"

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

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

data = mapping[ext](filename)

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

您认为用户是否决定列的数据类型-int、float或str?不,我的意思是说文件是否为csv、xls、pickle或其他格式。这确实是我的备份计划,但我想知道是否有某种内置方法可以做到这一点。无论如何谢谢你!看起来不太好哦那我就开始吧我想先确定一下谢谢