【只谈干货】python之os库常见用法

1. 导入

import os

2. os.path.join()

合并路径

model_dir = '/home/ys/project/'
num = 0
print(os.path.join(model_dir, 'image_%d.png' % num))

/home/ys/project/image_0.png

3. os.path.exists()

判断路径是否存在

print(os.path.exists(os.path.join(model_dir)))

存在：true
不存在： false

4. os.makedirs()

创建文件夹（常和os.path.exists()一起使用）

folder = '000'
if not os.path.exists(os.path.join(model_dir, folder)):
	os.makedirs(os.path.join(model_dir, folder))

5. os.walk()

遍历目录中的文件

folder_name = '/home/ys/project/rtw/datasets/'
for root, dirs, files in os.walk(folder_name, topdown=False):

root 所指的是当前正在遍历的这个文件夹的本身的地址
dirs 是一个 list ，内容是该文件夹中所有的文件夹(不包括子目录)
files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)

6. os.path.splitext()

分离文件的文件名和扩展名

image = 'image_01.png'
os.path.splitext(image)

(‘image_01’, ‘.png’)