python文件、字符串处理

1、删除、创建文件夹

import shutil
if os.path.exists(outimagePathfile):
    shutil.rmtree(outimagePathfile)
os.mkdir(outimagePathfile)

说明：outimagePathfile文件夹存在，则递归删除里面的文件，新建该文件夹；如果不存在则直接新建。

 

2、获得某文件夹下的所有文件

filenames = []
file_path = '%s*' % (image_dir)
matching_files = tf.gfile.Glob(file_path)
filenames.extend(matching_files)

说明：image_dir是'/xx/xx/xx/'格式的路径，filenames包含该路径下所有文件的名字（带路径）。

 

3、获得某文件夹下所有的图像

idList = os.listdir(filenames[i])  # 该文件夹下的所有图像
for imageIndex in idList:   #imageIndex是每个图像的名字（不带路径）
    if not ('ir' in imageIndex):   #图像名字中，如果没有'ir'字符则不进行后面的处理
        continue
    image_name = os.path.join(filenames[i], imageIndex) #获得路径+图像名字
    image = cv2.imread(image_name)
    image_out = image
    outname = os.path.join(outimagePathfile, imageIndex) #输出图像的路径+图像名字
    cv2.imwrite(outname, image_out)

4、拆分字符串

outimagefile = filenames[i].split('U8C3/', 1)
outimagePathfile = output_directory + outimagefile[1]

说明：将字符串filenames[i]以'U8C3/'为界分为两部分，outimagefile[1]即为后面那部分字符串。