python获取文件夹下所有文件,大家肯定都会想到os.listdir了,但是这种方法只能够获取到当前文件夹中的所有文件/文件夹,如果我想获得文件夹-文件夹-文件呢?
我的需求是只要文件夹中所有非文件夹,换句话来说对于我们的需求只有文件夹和非文件夹两种情况,那么直接联想到二叉树遍历。
import os
def ddd(root:os.path):
roots = os.listdir(root)
if roots is None:
return []
res = []
for i in roots:
a=os.path.join(root,i)
if os.path.isdir(a):
res+=ddd(a)
else:
res.append(a)
return res
ddd(r'C:\Users\gzddz\Desktop\dd')