python递归遍历

使用python对制定文件夹下制定后缀的文件进行遍历.

os.path.exists(path)

os.makedirs

os.path.join

os.path.isfile

(path, extension) = os.path.splitext(source_file)

# -*- coding: utf-8 -*-
__author__ = 'big_centaur'
#采用递归遍历的方式遍历图片
import os
from PIL import Image
# 要处理的目录
folder = 'datasets'
def recurve_opt(root_path):
for file in os.listdir(root_path):
target_file = os.path.join(root_path, file)
if os.path.isfile(target_file):
(path, extension) = os.path.splitext(target_file)
if extension == '.jpg' or extension == '.png':
# Do something 此处我用于把图片转为灰度图像
# image = Image.open(target_file).convert('L')
# image.save(target_file)
# print(target_file)
else:
recurve_opt(target_file)
def main():
recurve_opt(folder)
if __name__ == '__main__':
main()

# -*- coding: utf-8 -*-
__author__ = 'big_centaur'
#采用递归遍历的方式遍历图片
def recurve_opt(root_1, root_2):
if not os.path.exists(root_2):
os.makedirs(root_2)
for file in os.listdir(root_1):
source_file = os.path.join(root_1, file)
target_file = os.path.join(root_2, file)
if os.path.isfile(source_file):
(path, extension) = os.path.splitext(source_file)
if extension == '.jpg' or extension == '.png':
# shutil.copy(source_file, target_file)
# do something 此处我用于直接复制图像
else:
recurve_opt(source_file, target_file)
def main():
floder_1 = 'datasets/captcha/test_adjust'
floder_2 = 'datasets/captcha/test_adjust_remove_extar'
# floder_1:源文件夹 floder_2:目的文件 遍历源文件下的所有文件,按照原来的顺序存在目的文件夹中
recurve_opt(floder_1, floder_2)
if __name__ == '__main__':
main()