Python-读取目录所有文件的文件名，并可定制编号顺序保存到txt文件

转载于 2019-03-08 15:01:39 发布 · 3k 阅读

88 篇文章

订阅专栏

本文介绍了一个使用Python编写的脚本，该脚本能够遍历指定目录及其子目录下所有符合特定扩展名（如.jpg）的文件，并将文件名写入到一个txt文件中。此方法适用于需要对大量图片进行批量处理的场景，例如在搭建图像识别或计算机视觉项目时的数据预处理阶段。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

def ListFilesToTxt(dir, file, wildcard, recursion):
    exts = wildcard.split(" ")
    files = os.listdir(dir)
    files.sort(key=lambda x: int(x[:-4]))

    for name in files:
        fullname = os.path.join(dir, name)
        if (os.path.isdir(fullname) & recursion):
            ListFilesToTxt(fullname, file, wildcard, recursion)
        else:
            for ext in exts:
                if (name.endswith(ext)):
                    file.write(name + "\n")
                    break


def Test():
    dir = r"C:\Users\87419\Desktop\VAE1\data\trainA"  # 读入
    outfile = "label.txt"  # 写入
    # wildcard = ".jpg .txt .exe .dll .lib"  # 要读取的文件类型
    wildcard = ".jpg"

    file = open(outfile, "w")
    if not file:
        print("cannot open the file %s for writing" % outfile)

    ListFilesToTxt(dir, file, wildcard, 1)

    file.close()

if __name__ == '__main__':
    Test()