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

朱雀 2022-03-11 19:51 331阅读 0赞

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()

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTM4NjY2_size_16_color_FFFFFF_t_70][]

[https://blog.csdn.net/CVAIDL/article/details/79470096?utm\_source=blogxgwz0][https_blog.csdn.net_CVAIDL_article_details_79470096_utm_source_blogxgwz0]  mark

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTM4NjY2_size_16_color_FFFFFF_t_70]: /images/20220311/31aac4528ccf4e048da4e9ca5d47b577.png
[https_blog.csdn.net_CVAIDL_article_details_79470096_utm_source_blogxgwz0]: https://blog.csdn.net/CVAIDL/article/details/79470096?utm_source=blogxgwz0