把文件名写进txt文件中

最新推荐文章于 2024-08-03 14:24:04 发布

summer2day

最新推荐文章于 2024-08-03 14:24:04 发布

阅读量1.3k

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/summer2day/article/details/83064652

版权

python 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

在用深度学习跑实验时，对数据集的处理需要把文件名写进一个txt文档中，结果是每个名字占一行

import os
def select(path):
    files=os.listdir(path)#列出所有文件
    for file in files:
        filename=os.path.splitext(file)[0]#分割出文件名
        sufix=os.path.splitext(file)[1]#分割出后缀
        if(sufix=='.xml'):
            with open("test.txt","a") as f:   
                f.writelines(filename+'\n')
                           
if __name__ == "__main__":
    path=("D:/NWPU VHR-10 dataset/测试集/Annotations")
    select(path)