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

要求很简单

比如我在做机器学习实验的时候，实验结果的保存路径是'runs/exp'。

这样就会出现一个问题：当我第二次运行程序的时候，如果我忘记更改代码中的路径名或者清除上次实验结果，这次的实验结果会和上次实验结果混在一起，很头疼。

所以我希望在下次实验的时候代码自动创建一个新的目录，如：'runs/exp2'，来保存新的实验结果。

代码：

粘贴复制，直接照着下面用就行。

下面的函数是我从yolo5里复制过来的，代码质量很高，鲁棒性很高，可以放心使用。

1. from pathlib import Path
   
2. import re
   
3. import glob
   
4. 
   
5. def increment_path(path, exist_ok=False, sep='', mkdir=True):
   
6.     """
   
7.     Increment file or directory path, i.e. runs/exp --> runs/exp{sep}2, runs/exp{sep}3, ... etc.
   
8.     :param path: file or directory path to increment
   
9.     :param exist_ok: existing project/name ok, do not increment
   
10.     :param sep: separator for directory name
    
11.     :param mkdir: create directory
    
12.     :return: incremented path
    
13.     """
    
14.     path = Path(path)  # os-agnostic
    
15.     if path.exists() and not exist_ok:
    
16.         suffix = path.suffix
    
17.         path = path.with_suffix('')
    
18.         dirs = glob.glob(f"{path}{sep}*")  # similar paths
    
19.         matches = [re.search(rf"%s{sep}(\d+)" % path.stem, d) for d in dirs]
    
20.         i = [int(m.groups()[0]) for m in matches if m]  # indices
    
21.         n = max(i) + 1 if i else 2  # increment number
    
22.         path = Path(f"{path}{sep}{n}{suffix}")  # update path
    
23.     dir_ = path if path.suffix == '' else path.parent  # directory
    
24.     if not dir_.exists() and mkdir:
    
25.         dir_.mkdir(parents=True, exist_ok=True)  # make directory
    
26.     return path
    

复制代码

使用方法

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

复制代码

效果如下：

因为当前目录下已经有了一个’runs/exp’，所以函数increment_path会自动创建增量目录’runs\exp2’并返回新的路径名。

好了，上述就是最常用的使用方法，接下来是我自己的笔记。

1、函数increment_path(path, exist_ok=False, sep='', mkdir=True)的参数含义如下：

• path: 表示要增量的目录，如：‘runs/exp’
• exist_ok: 若为True，则不增量路径，直接使用原来的路径。
• sep: 目录名的分隔符，比如sep=’_’，则返回’runs/exp_2’
• mkdir: 若为False，则只返回路径名，不会自动创建目录。
  

2、其他笔记

pathlib.Path

1. from pathlib import Path
   
2. 
   
3. path = Path(path)
   
4. path.suffix  # 表示获取文件后缀名
   
5. path.with_suffix('')  # 表示更改文件后缀名为''
   
6. path.stem  # 返回文件名，不带后缀

复制代码

glob.glob

[size=1em]

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

复制代码