使用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列表存儲的是所有的完整的文件名路徑