Python3 將一個文件夾中的文件按一定數量分組批量復制到新的文件夾

需求：一個文件夾中有多個jpg文件和ans文件，將jpg文件按一定數量分組復制新的文件夾，

如果原文件夾中有同名的ans文件則一同復制到新文件夾中

參考資料地址：https://www.jb51.net/article/167794.htm

代碼：

#!/usr/bin/env python
# encoding: utf-8

import os, glob
from shutil import copy


def copyFile(file_path, save_dir, count, dir_name):
        #i是用來計算文件的數量，k是用來計算新建文件夾的數量
        i = 0
        k = 0

        #如果目錄不存在，則創建
        if not os.path.isdir(save_dir):
                os.makedirs(save_dir)

        #通過glob.glob來獲取原始路徑下，所有'.jpg'文件
        imageList = glob.glob(os.path.join(file_path, '*.jpg'))

        for allImgDir in imageList:
                print(allImgDir)
                #獲取文件名稱（包括后綴）
                imgDir = os.path.basename(allImgDir)
                print(imgDir)
                #更改jpg文件后綴為ans
                (temp1, temp2) = os.path.splitext(imgDir)
                ansDir = temp1 + '.ans'

                #拼接路徑與文件名
                from_imgPath = file_path+'/'+imgDir
                from_ansPath = file_path+'/'+ansDir
                #新建的文件夾
                to_path = save_dir + "\\" + dir_name + str(k)

                # 如果 to_path 目錄不存在，則創建
                if not os.path.isdir(to_path):
                        os.makedirs(to_path)
                copy(from_imgPath, to_path)
                if os.path.exists(from_ansPath):
                        #將ans文件遍歷復制到目標文件夾中
                        copy(from_ansPath, to_path)
                i += 1
                if((i%count) == 0):
                        print('新建一個文件夾')
                        k += 1

if __name__ == '__main__':
        inputfile = input('請輸入想要拆分的文件夾路徑（如：E:/input/）：')
        outfile = input('請輸入想保存的根目錄（如：E:/666/）：')
        count = int(input('請輸入分組的數量（如：2）：'))
        dir_name = input('請輸入想保存的文件名字（如：圖片）：')
        #指定找到文件后，另存為的文件夾路徑
        save_dir = os.path.abspath(outfile)
        #指定文件的原始路徑
        file_path = os.path.abspath(inputfile)
        copyFile(file_path, save_dir, count, dir_name)