利用Python对文件进行各种操作

遍历文件夹输出文件夹中所有文件名称

实现：利用os.listdir()

e.g

for idx in os.listdir(kpt_folder):
    print(idx)  # 这里的idx即为文件夹中文件的名称

寻找文件夹中包含指定关键词的文件

实现：利用os.listdir()

e.g

# 根据指定关键词查找文件
keywords = 'your_keywords'
        for a in os.listdir(current_img_dir):
            if keywords in a:
                current_img = osp.join(current_img_dir, a)

当有两个关键词时，使用的是:

if keywords1 in a or keywords2 in a:

将文件名及拓展名分离

实现：利用osp.splitext()

e.g

(filename, extension) = osp.splitext(img_names)

指定字符将文件名分割

实现：利用.split()

e.g

# 这里的`_`就是指定的字符，[0]代表取分割后的第一部分
img_id = filename.split('_')[0]

若不存在文件夹则创建

实现：利用os.path.exists()及os.makedirs()

e.g

    if not osp.exists(current_save_image_dir):
        os.makedirs(current_save_image_dir)