提取目录下所有文件的名字到txt文档（python）

file_name.py

这里提取的是所有xml格式的文件名

from glob import glob

ftrainval = open('C:/Users/huan4/Desktop/Python/VOC2007/ImageSets/trainval.txt', 'a')    #保存文件的路径
total_files = glob("C:/Users/huan4/Desktop/Python/tool/Annotations/*.xml")   #要提取的xml文件路径
total_files = [i.split("/")[-1].split(".xml")[0] for i in total_files]
for file in total_files:
    ftrainval.write(file + "\n")

ftrainval.close()