python获取某文件夹下所有指定后缀的文件

python获取某文件夹下所有指定后缀的文件。在某个文件夹下，可能包含几种文件类型的数据，如.png、.txt、.json，有时只需要获取其中的某类文件进行处理。

相关函数:

• os.path.splitext()：分离文件名与扩展名
• os.listdir()：查找本目录下所有文件

代码：

import os
datanames = os.listdir(train_dir)
for dataname in datanames:
    if os.path.splitext(dataname)[1] == '.json':#目录下包含.json的文件
        print(dataname)