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.