使用python怎么對(duì)磁盤目錄進(jìn)行遍歷

這篇文章給大家介紹使用python怎么對(duì)磁盤目錄進(jìn)行遍歷，內(nèi)容非常詳細(xì)，感興趣的小伙伴們可以參考借鑒，希望對(duì)大家能有所幫助。

深度遍歷

遞歸

import os

def get_files(path):
  # 判斷路徑是否存在,如果不存在，函數(shù)直接結(jié)束
  if not os.path.exists(path):
    print('路徑不存在')
    return
  # 判斷路徑是否為文件夾
  if not os.path.isdir(path):
    print('路徑是一個(gè)文件')
    return
  # 這時(shí)候，路徑是一個(gè)文件夾
  # 獲取文件夾中文件或文件夾的名稱
  file_list = os.listdir(path)
  # 遍歷文件夾
  for filename in file_list:
    # 拼接路徑，獲取每個(gè)次級(jí)目錄下的文件路徑
    subpath = os.path.join(path,filename)
    if os.path.isfile(subpath):
      if os.path.splitext(subpath)[1] == '.py':
        print('python文件：{}'.format(subpath))
    else:
      # 如果filename是文件夾，則調(diào)用函數(shù)繼續(xù)遍歷
      get_files(subpath)

用棧來(lái)遍歷磁盤

棧的特點(diǎn)：先進(jìn)后廚，后進(jìn)先出
原理：path第一次被pop刪除后返回path，遍歷目錄下的文件，如果遇到文件夾追加到列表中，pop是刪除最后一位的元素，每次又遍歷最后一位的文件夾，所以每一輪都會(huì)將次級(jí)目錄下的文件夾遍歷完成之后再遍歷下個(gè)次級(jí)目錄

import os

def get_files(path):
  # 判斷路徑是否存在
  if not os.path.exists(path):
    print('路徑不存在')
    return
  if not os.path.isdir(path):
    print('路徑是一個(gè)文件夾')
    return
  # 創(chuàng)建一個(gè)列表作為棧
  stack = [path]
  # 取出棧中的元素
  while len(stack) != 0:
    path = stack.pop()
    file_list = os.listdir(path)
    for filename in file_list:
      subpath = os.path.join(path,filename)
      if os.path.isfile(subpath):
        print('python文件：{}'.format(subpath))
      else:
        stack.append(subpath)

廣度遍歷磁盤

用隊(duì)列遍歷磁盤

import os
import collections


def get_py_file(path):
  # 判斷路徑是否存在
  if not os.path.exists(path):
    print('路徑不存在')
    return
    # 判斷路徑是否是文件夾
  if os.path.isfile(path):
    print('路徑是文件')
    return
    # path是一個(gè)文件夾

  # 定義一個(gè)空對(duì)列
  queue = collections.deque()
  queue.append(path)
  while len(queue) != 0:
    # 從隊(duì)列中獲取第一個(gè)元素
    path = queue.popleft()
    # 獲取目錄下的所有內(nèi)容
    filelist = os.listdir(path)
    # 遍歷
    for filename in filelist:
      # 拼接
      filepath = os.path.join(path, filename)
      if os.path.isfile(filepath):
        if os.path.splitext(filepath)[1] == '.py':
          print(filepath)
      else:
        queue.append(filepath)

關(guān)于使用python怎么對(duì)磁盤目錄進(jìn)行遍歷就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺(jué)得文章不錯(cuò)，可以把它分享出去讓更多的人看到。