python 读取一个文件夹下的所jpg文件保存到txt中

最近需要使用统计一个目录下的所有文件，使用python比较方便，就整理了一下代码。

 

import os

def gci(filepath):
    files = os.listdir(filepath)
    for fi in files:
        fi_d = os.path.join(filepath,fi)
        if os.path.isdir(fi_d):
            gci(fi_d)
        else:
            str = os.path.join(filepath,fi_d)+'\n' 
            suffix = str[-5:]
            print(suffix)
            if(suffix == ".jpg\n"):
                print(str ,'is a jpg file' + '\n')
                list_txt.writelines(str)
            else:
                print(str,'is not a jpg file' + '\n')


# recursion traverse
list_txt = open('list.txt', 'w')
path = r'D:\work\errorPic'
print(path)
gci(path)
list_txt.close()

 在使用过程中发现使用记事本可以正常显示txt中的中文，使用notepad++查看时中文显示为乱码，解决办法：

将上述代码中的一行改为下面一行：

list_txt = open('list.txt', 'w', encoding='utf-8')

也就是在开始写文件时就指定编码格式为utf-8.