Python 8行代码实现查找目录下所有子文件, 子目录

113 阅读 0 评论 75 点赞

我是靠谱客的博主深情大碗，这篇文章主要介绍Python 8行代码实现查找目录下所有子文件, 子目录，现在分享给大家，希望可以做个参考。

代码如下, findall_files 表示获得目标文件夹下所有子文件 (包括任意深度的子目录中的), findall_subdirs 表示获得目标文件夹下所有子目录:

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# file_finder.py
import os


def findall_files(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend([os.path.join(root, f) for f in files])
    
    return collector


def findall_subdirs(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend(os.path.join(root, d) for d in dirs)
    
    return collector

注意: os.path.join 在 windows 上是通过反斜杠连接的, 比如我的 root 是 “…/A”, 我的 file 是 “B.txt”, 则连接的结果是 “…/AB.txt”.

如果想统一用正斜杠的话, 可以用下面改良后的代码:

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# file_finder.py
import os


def findall_files(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend([os.path.join(root, f) for f in files])
    
    return [x.replace('\', '/') for x in collector]


def findall_subdirs(main_dir: str) -> list:
    collector = []
    
    for root, dirs, files in os.walk(main_dir):
        collector.extend(os.path.join(root, d) for d in dirs)
    
    return [x.replace('\', '/') for x in collector]