Python 文件命令总结

读取 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)