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.