統計文件夾中圖片的個數

條件：有一個文件夾，裏面不僅有子文件夾，還有一些其他格式的文件。子文件夾裏面是圖片， 現在需要統計每個子文件夾中圖片的個數。

import os
path = 'D:/testin/common-mobile-web-app-icons/'  #父文件夾路徑
all_folds = os.listdir(path)   #解析出父文件夾中所有的文件名稱，並以列表的格式輸出，
#例如['add','common-mobile-web-app-icons.zip', '新建 Microsoft Word 文檔.docx', 'arrow_down']
l = []
for i in range(len(all_folds)):
    fold_path = os.path.join(path,all_folds[i])  #將父文件夾路徑加上子文件的名稱，例如：'D:/testin/common-mobile-web-app-icons/add'
    if os.path.isdir(fold_path):   #判斷該文件是否爲文件夾
        count_fold = len(os.listdir(fold_path))
        #print(all_folds[i],count_fold)
        l.append((all_folds[i],count_fold))  #得到列表，列表裏面是數組，數組裏面是文件名稱和該文件名稱文件夾中圖片個數
#print(l)
dic_file = dict(l)  #數組轉成字典
#dic_file
txt_file = os.getcwd()+'\count.txt'  #os.getcwd()得到當前路徑，並在當前路徑下建一個txt文本文件
out = open(txt_file,'w')  #打開文本文件
for i in  dic_file:  #循環字典的鍵
    out.write(i)  #寫入鍵，既文件夾名稱
    out.write(': ')  
    out.write(str(dic_file[i]))  #寫入值，既文件夾名稱下的圖片個數
    out.write('\n')  #換行
out.close()  #關閉txt文本文件