os.path.splitext()的用法

这个函数的意思是将文件的格式分开，比如输入为data/p1ch4/image-cats/add.png，那么输出为(‘data/p1ch4/image-cats/add’, ‘.png’)，返回的是一个元组。
我们一般的用法是
if os.path.splitext(‘data/p1ch4/image-cats/add.png’)[-1] == ‘.png’。来判断这个文件是什么文件，是不是我们需要的文件。

这个方法也通常和os.listdir()一起使用。
os.listdir函数是将路径下的所有文件读入，结果是一个列表，比如
data_dir = ‘…/data/p1ch4/image-cats/’
os.listdir(data_dir)
​结果为：[‘cat1.png’, ‘cat2.png’, ‘cat3.png’]
当然，我这里全是png格式的文件。
流程就是。先用os.listdir把文件名全部读取，然后再用os.splitext判断哪些格式的文件是我们需要的。