我写函数与一些数据一起运行。我从读取文件中的数据开始,将其放入结构中,然后进行一些计算。 我的问题 - 传入的文件可以是txt,csv或xls(xlsx)。 对于不同类型的文件,我需要使用不同的方法。
所以,我的问题是 - 如何检查传入文件的类型。
现在我使用下一种方法:
def main(filename):
if '.txt' in filename:
# use read_txt
elif '.xls' in filename:
print('there is no txt method')
else:
print('etc.')
但它并不完美:例如,如果我创建文件My.txt.xls
则失败。
我想到try
并逐个使用所有的阅读功能,但看起来有一些更明显的方式......