python把文件夹/子文件夹内的图片导到指定文件夹

<span style="font-family: Arial, Helvetica, sans-serif;"># -*- coding: utf-8 -*-</span>

#把path文件夹下以及其子文件下的所有png图片移动（不是复制）到new_path

import os
import shutil
path = 'E:/Lin/data_1208/third'
new_path = 'E:/Lin/data_1208/test'

for root, dirs, files in os.walk(path):
    if len(dirs) == 0:
        for i in range(len(files)):
            if files[i][-3:-1] == 'png':
                file_path = root+'/'+files[i]
                new_file_path = new_path+ '/'+ files[i]
                shutil.move(file_path,new_file_path)

早上对数据绘图，绘完图后保存在了数据所在的文件夹，之后需要把所有图片都整到同一个文件夹里，手动拷贝太复杂，于是就写了个脚本。

path是原来的文件夹，new_path是图片的目标文件夹，程序的目的就是把path文件夹内的所有图片（可以穿过子目录）都导到new_path里