使用Python获取文件夹下面所有指定后缀文件名的文件路径列表

一行代码搞定

获取文件夹下面的所有xml文件

import glob
train_path = 'D:\\AI_Kaggle_Study\voc2coco\\mask\\train\\'
xml_list = glob.glob(train_path + "/*.xml")
print(xml_list)

或者

import os
xml_list = [os.path.join(train_path, i) for i in os.listdir(train_path) if i[-4:] == '.xml']
print(xml_list )

或者

import os
xml_list = [os.path.join(train_path, i) for i in os.listdir(train_path) if i.endswith('.xml')]
print(xml_list )

或者

import os
xml_list = [os.path.join(train_path, i) for i in os.listdir(train_path) if os.path.splitext(i)[-1] == '.xml']
print(xml_list)

注意：上面的xml_list列表存储的是所有的完整的文件名路径