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)