經常有需要掃描目錄，對檔案做批量處理的需求，所以對目錄處理這塊做了下學習和總結。Python 中掃描目錄有兩種方法：os.listdir 和 os.walk。

一、os.listdir 方法

os.listdir() 方法用於返回指定的目錄下包含的檔案或子目錄的名字的列表。這個列表以字母順序。其得到的是僅當前路徑下的檔名，不包括子目錄中的檔案，如果需要得到所有檔案需要遞迴。 它也不包括 '.' 和 '..' 即使它在目錄中。

語法格式如下：

os.listdir(path)

例項程式碼

def list_dir(file_dir):
  '''
    通過 listdir 得到的是僅當前路徑下的檔名，不包括子目錄中的檔案，如果需要得到所有檔案需要遞迴
  '''
  print'\n\n<><><><><><> listdir <><><><><><>'
  print "current dir : {0}".format(file_dir)
  dir_list = os.listdir(file_dir)
  for cur_file in dir_list:
    # 獲取檔案的絕對路徑
    path = os.path.join(file_dir,cur_file)
    if os.path.isfile(path): # 判斷是否是檔案還是目錄需要用絕對路徑
      print "{0} : is file!".format(cur_file)
    if os.path.isdir(path):
      print "{0} : is dir!".format(cur_file)
      list_dir(path) # 遞迴子目錄

二、os.walk 方法

os.walk() 方法用於通過在目錄樹中游走輸出在目錄中的檔名，向上或者向下。是一個簡單易用的檔案、目錄遍歷器，可以幫助我們高效的處理檔案、目錄方面的事情。

語法格式如下：

os.walk(top[,topdown=True[,onerror=None[,followlinks=False]]])

方法引數說明：

• top：要遍歷的目錄的路徑
• topdown：可選，如果為 True，則優先遍歷 top 目錄，以及 top 目錄下的每一個子目錄，否則優先遍歷 top 的子目錄，預設為 True
• onerror: 可選， 需要一個 callable 物件，當 walk 異常時呼叫
• followlinks：可選， 如果為 True，則會遍歷目錄下的快捷方式(linux 下是 symbolic link)實際所指的目錄，預設為 False
• args：包含那些沒有 '-' 或 '--' 的引數列表

返回值: 三元組 (root,dirs,files)

• root ：所指的是當前正在遍歷的目錄的地址
• dirs ：當前資料夾中所有目錄名字的 list (不包括子目錄)
• files ：當前資料夾中所有的檔案 (不包括子目錄中的檔案)

例項

def work_dir(file_dir):
  print'\n\n<><><><><> work dir <><><><><>'
  for root,files in os.walk(file_dir):
    print'\n========================================'
    print "root : {0}".format(root)
    print "dirs : {0}".format(dirs)
    print "files : {0}".format(files)
​
    for file in files:
      try:
        print'-----------------------------------'
        
        file_name = os.path.splitext(file)[0]
        file_suffix = os.path.splitext(file)[1]
        file_path = os.path.join(root,file)
        file_abs_path = os.path.abspath(file)
        file_parent = os.path.dirname(file_path)
​
        print "file : {0}".format(file)
        print "file_name : {0}".format(file_name)
        print "file_suffix : {0}".format(file_suffix)
        print "file_path : {0}".format(file_path)
        print "file_abs_path : {0}".format(file_abs_path)
        print "file_parent : {0}".format(file_parent)
        
      except Exception,e:
        print "Exception",e

三、其他跟檔案相關的常用方法

os.path.splitext()：分離檔名和副檔名

file = "file_test.txt"
file_name = os.path.splitext(file)[0] # 輸出：file_test
file_suffix = os.path.splitext(file)[1] # 輸出：.txt

os.path.exists()：判斷檔案或目錄是否存在

os.path.isfile()：判斷是否是檔案

os.path.isdir()：判斷是否是目錄

os.path.dirname()：獲取當前檔案所在的目錄，即父目錄

os.makedirs()：建立多級目錄

os.mkdir()：建立單級目錄

os.path.getsize()：獲取檔案大小

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。