python怎样实现遍历磁盘目录?一些Python新手对于如何便利磁盘不是很清楚,因此这篇文章就给大家介绍一下使用深度遍历,用栈遍历和广度遍历磁盘,需要的朋友可以参考学习。
import os
def get_files(path):
# 判断路径是否存在,如果不存在,函数直接结束
if not os.path.exists(path):
print('路径不存在')
return
# 判断路径是否为文件夹
if not os.path.isdir(path):
print('路径是一个文件')
return
# 这时候,路径是一个文件夹
# 获取文件夹中文件或文件夹的名称
file_list = os.listdir(path)
# 遍历文件夹
for filename in file_list:
# 拼接路径,获取每个次级目录下的文件路径
subpath = os.path.join(path,filename)
if os.path.isfile(subpath):
if os.path.splitext(subpath)[1] == '.py':
print('python文件:{}'.format(subpath))
else:
# 如果filename是文件夹,则调用函数继续遍历
get_files(subpath)
栈的特点:先进后厨,后进先出
原理:path第一次被pop删除后返回path,遍历目录下的文件,如果遇到文件夹追加到列表中,pop是删除最后一位的元素,每次又遍历最后一位的文件夹,所以每一轮都会将次级目录下的文件夹遍历完成之后再遍历下个次级目录
import os
def get_files(path):
# 判断路径是否存在
if not os.path.exists(path):
print('路径不存在')
return
if not os.path.isdir(path):
print('路径是一个文件夹')
return
# 创建一个列表作为栈
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)
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是一个文件夹
# 定义一个空对列
queue = collections.deque()
queue.append(path)
while len(queue) != 0:
# 从队列中获取第一个元素
path = queue.popleft()
# 获取目录下的所有内容
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)
以上就是关于实现python遍历磁盘目录的三种方法介绍,对大家了解遍历具有一定的参考价值,需要的朋友可以看看。
文本转载自脚本之家免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。
长按识别二维码并关注微信
更方便到期提醒、手机管理