Python 8行代码实现查找目录下所有子文件, 子目录 |
您所在的位置:网站首页 › python查询当前目录所有文件 › Python 8行代码实现查找目录下所有子文件, 子目录 |
代码如下, findall_files 表示获得目标文件夹下所有子文件 (包括任意深度的子目录中的), findall_subdirs 表示获得目标文件夹下所有子目录: # 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”, 则连接的结果是 “…/A\B.txt”. 如果想统一用正斜杠的话, 可以用下面改良后的代码: # 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] |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |