问题描述:给定一个路径,该路径下仅有一层文件夹,遍历得到该路径下的所有文件
#-*- coding: UTF-8 –*-
import os
path = "G:\\BigKnowledge\\"
files = os.listdir(path)
s = []
for file in files: #遍历第一层path
if os.path.isdir(path + file): #判断是否文件夹
subfiles = os.listdir(path + file)
for subfile in subfiles:#遍历第二层path
s.append(subfile)
else:
s.append(file)
print(s)
print(len(s))
问题升华:给定一个路径,遍历得到该路径下的所有文件
解决办法:引入深度搜索
# -*- coding: UTF-8 –*-
import os
path = "G:\\BigKnowledge"
s = []
def DFS(path):
files = os.listdir(path)
for file in files: #遍历第一层path
if os.path.isdir(path + '\\'+ file): #判断是否文件夹
DFS(path + '\\'+ file)
else:
s.append(file)
DFS(path)
print(s)
print(len(s))