图片处理之图片相同的后缀分类

from PIL import Image
import os
import os.path
import numpy as np
import cv2
src_path = 'C:/Users/jiabin.ouyang/Desktop/test_picture/'
save_path1 = r"C:/Users/jiabin.ouyang/Desktop/picture_1_sort/"
save_path2 = r"C:/Users/jiabin.ouyang/Desktop/picture_2_sort/"
save_path3 = r"C:/Users/jiabin.ouyang/Desktop/picture_3_sort/"
save_path4 = r"C:/Users/jiabin.ouyang/Desktop/picture_4_sort/"
save_path5 = r"C:/Users/jiabin.ouyang/Desktop/picture_5_sort/"
save_path6 = r"C:/Users/jiabin.ouyang/Desktop/picture_6_sort/"
listimg = os.listdir(src_path)
for i in range(len(listimg)):
    src_path_img = src_path + listimg[i]
    a = listimg[i].split('_')  # 按"_"进行切分
    b = a[0]+'_'+a[1]
    if a[1] == '1.png':
        save_path_img = save_path1+b      #+'.png'
        image = cv2.imread(src_path_img)
        cv2.imwrite(save_path_img,image)
    elif a[1]== '2.png':
        save_path_img =save_path2+b      #+'.png'
        image=cv2.imread(src_path_img)
        cv2.imwrite(save_path_img,image)
    elif a[1] == '3.png':
        save_path_img = save_path3 + b  # +'.png'
        image = cv2.imread(src_path_img)
        cv2.imwrite(save_path_img, image)
    elif a[1] == '4.png':
        save_path_img = save_path4 + b  # +'.png'
        image = cv2.imread(src_path_img)
        cv2.imwrite(save_path_img, image)
    elif a[1] == '5.png':
        save_path_img = save_path5 + b  # +'.png'
        image = cv2.imread(src_path_img)
        cv2.imwrite(save_path_img, image)
    elif a[1] == '6.png':
        save_path_img = save_path6 + b  # +'.png'
        image = cv2.imread(src_path_img)
        cv2.imwrite(save_path_img, image)
    else:
        print('more!')