python提取文件夾中的所有png文件並保存

此python腳本由於提取文件夾中所有子文件夾裏的png文件，並將他們保存於另一新文件夾，常用於LabelMe的Json文件處理

import os
import shutil
 
path = '/home/zhang/桌面/Annotation'
new_path = '/home/zhang/文檔/out_11_labeled'
count = os.listdir(path)
for j in range(1,len(count)+1):
    for root, dirs, files in os.walk(path):
        if len(dirs) == 0:
            for i in range(len(files)):
                print("i=",i)
                if files[i].find('label.png')!=-1:
                    shutil.copy(os.path.join(path+'/out11_'+str(j).zfill(1)+'_json/',files[i]), os.path.join(new_path,'out11_'+str(j).zfill(1)+'.png'))

# path爲包含所有要提取的png文件的原文件夾名
# png文件名根據‘out11_’修改