經常有需要掃描目錄，對文件做批量處理的需求，所以對目錄處理這塊做了下學習和總結。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, dirs, 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()：獲取文件大小

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