深度遍历
递归
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是删除最后一位的元素,每次又遍历最后一位的文