处理数据的代码~python

1、判断两个文件夹的数据是不是一样多的。查不到原图或者查不到label的进行删除

import os
trainAdir='./trainA/'
trainBdir='./trainB/'

for fi in os.listdir(trainBdir):
        if not os.path.exists(trainAdir+'1'+fi[1:]):
            os.remove(trainBdir+fi)
            print (fi,'....................................')
            
for fi in os.listdir(trainAdir):
        if not os.path.exists(trainBdir+'0'+fi[1:]):
            os.remove(trainAdir+fi)
            print (fi,'.....!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!')