1. 子文件夾的獲取
2. 文件的複製移動
# -*- coding:utf-8 -*-
import os
from shutil import copy2
allDir="D:\\pics\\易烊千璽"
trainDir = "C:\\Users\\WEIJIE\\Desktop\\train_data\\0"
dir=os.listdir(allDir) #獲取所有子文件夾
index=0
for i in dir:
folder=os.path.join(allDir, i)
#trainfolder=os.path.join(trainDir, i)
trainfiles = os.listdir(folder)
num_train = len(trainfiles)
print("num_train: " + str(num_train))
index_list = list(range(num_train))
for i in index_list:
fileName = os.path.join(folder, trainfiles[i])
outfile=os.path.join(trainDir, str(index)+'_'+trainfiles[i])
copy2(fileName,outfile) #複製文件
index+=1