读取 Excel:
import xlrd
data = xlrd.open_workbook('gt.xlsx')
table = data.sheet_by_name('Sheet1')
for i in range(table.nrows):
imgname, gt = table.row_values(i)[0], table.row_values(i)[1]
新建、删除:
import os
# 创建文件
with open('a.txt', 'w') as f:
f.write('hello world')
# 读取文件
with open('a.txt', 'r') as f:
text = f.read() # f.readlines()
# 删除文件
os.remove(path)
# 创建文件夹
# 区别:若 dir_path 的父文件夹不存在,则 mkdir 创建失败,而 mkdirs 会连同父文件夹一同创建;
os.mkdir(dir_path)
os.mkdirs(dir_path)
# 删除文件夹
os.rmdir(dir_path)
移动、复制:
import os, glob, shutil
src_dir = './datasets'
dst_dir = './new_dataset'
filelist = glob.glob('./datasets/*/*.jpg') # 当前目录的二级目录下所有JPG图像文件的路径
for file in filelist:
filename = os.split(file)[-1]
shutil.copyfile(file, os.path.join(dst_dir, filename)) # 复制文件
# shutil.move(file, os.path.join(dst_dir, filename)) # 移动文件
路径、搜索:
import os, glob
# 获取文件后缀名(.jpg, .png, .txt ...)
suffix = os.path.splitext(filepath)[-1]
# 获取文件名(不含路径)或路径
filename = os.path.split(filepath)[-1]
dir = os.path.split(filepath)[0]
# 连接路径、连接路径与文件名
filepath = os.path.join(dir, filename)
# 判断路径是否存在
if not os.path.exists(filepath):
os.mkdir(filepath) # 不存在则创建
# 切换路径
os.chdir(new_dir)
# 显示文件夹下所有文件(或符合条件的文件)
filelist = os.listdir(dir)
jpglist = glob.glob(os.path.join(dir, '*.jpg'))
# 显示文件所在的文件夹路径
os.path.dirname(filepath)