Python创建增量目录的代码实例

from pathlib import Path
import re
import glob

def increment_path(path, exist_ok=False, sep='', mkdir=True):
    """
    Increment file or directory path, i.e. runs/exp --> runs/exp{sep}2, runs/exp{sep}3, ... etc.
    :param path: file or directory path to increment
    :param exist_ok: existing project/name ok, do not increment
    :param sep: separator for directory name
    :param mkdir: create directory
    :return: incremented path
    """
    path = Path(path)  # os-agnostic
    if path.exists() and not exist_ok:
        suffix = path.suffix
        path = path.with_suffix('')
        dirs = glob.glob(f"{path}{sep}*")  # similar paths
        matches = [re.search(rf"%s{sep}(\d+)" % path.stem, d) for d in dirs]
        i = [int(m.groups()[0]) for m in matches if m]  # indices
        n = max(i) + 1 if i else 2  # increment number
        path = Path(f"{path}{sep}{n}{suffix}")  # update path
    dir_ = path if path.suffix == '' else path.parent  # directory
    if not dir_.exists() and mkdir:
        dir_.mkdir(parents=True, exist_ok=True)  # make directory
    return path

path = 'runs/exp'  # 定义你的路径名，比如我的是'runs/exp'
path = increment_path(path)
print(path)  # 打印输出 runs\exp2

from pathlib import Path

path = Path(path)
path.suffix  # 表示获取文件后缀名
path.with_suffix('')  # 表示更改文件后缀名为''
path.stem  # 返回文件名，不带后缀

import glob

glob.glob(f"./data/*.jpg")  # 找到./data/目录下所有.jpg后缀的文件