Python os庫常用操作代碼匯總

更新時間：2020年11月03日 11:51:19 作者：臨淵

這篇文章主要介紹了Python os庫常用操作代碼匯總,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

Python自動的os庫是和操作系統(tǒng)交互的庫，常用的操作包括文件/目錄操作，路徑操作，環(huán)境變量操作和執(zhí)行系統(tǒng)命令等。

文件/目錄操作

獲取當(dāng)前目錄(pwd): os.getcwd()
切換目錄(cd): os.chdir('/usr/local/')
列出目錄所有文件(ls)：os.listdir('/usr/local/')
創(chuàng)建目錄(mkdir)：os.makedirs('/usr/local/tmp')
刪除目錄(rmdir)：os.removedirs('/usr/local/tmp') # 只能刪除空目錄，遞歸刪除可以使用import shutil;shutil.rmtree('/usr/local/tmp')
刪除文件(rm)：os.remove('/usr/local/a.txt')
遞歸遍歷目錄及子目錄：os.walk()

示例：遍歷/usr/local目錄及子下所有文件和目錄，并組裝出每個文件完整的路徑名

import os
for root, dirs, files in os.walk("/usr/local", topdown=False):
  for name in files:
    print('文件：', os.path.join(root, name))
  for name in dirs:
    print('目錄：', os.path.join(root, name))

路徑操作

當(dāng)前Python腳本文件：__file__
獲取文件所在路徑：os.path.basename(__file__) # 不含當(dāng)前文件名
獲取文件絕對路徑：os.path.abspath(__file__) # 包含當(dāng)前文件名
獲取所在目錄路徑：os.path.dirname(__file__)
分割路徑和文件名：`os.path.split('/usr/local/a.txt') # 得到一個[路徑,文件名]的列表
分割文件名和擴(kuò)展名：os.path.splitext('a.txt') # 得到['a', '.txt']
判斷路徑是否存在：os.path.exists('/usr/local/a.txt')
判斷路徑是否文件：os.path.isfile('/usr/local/a.txt')
判斷路徑是否目錄：os.path.isdir('/usr/local/a.txt')
組裝路徑：os.path.join('/usr', 'local', 'a.txt')

示例：獲取項目根路徑和報告文件路徑

假設(shè)項目結(jié)構(gòu)如下

project/
data'
reports/
report.html
testcases/
config.py
run.py

在run.py中獲取項目的路徑和report.html的路徑

# filename: run.py
import os

base_dir = os.path.dirname(__file__) # __file__是run.py文件，os.path.dirname獲取到其所在的目錄project即項目根路徑
report_file = os.path.join(base_dir, 'reports', 'report.html') # 使用系統(tǒng)路徑分隔符('\')連接項目根目錄base_dir和'reports'及'report.html'得到報告路徑
print(report_file)

環(huán)境變量操作