使用python3.x实现文件遍历

原创于 2022-05-01 23:31:19 发布 · 1.1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python

python 专栏收录该内容

28 篇文章

订阅专栏

该博客介绍了一段Python代码，用于在指定文件夹及其子文件夹中查找所有以特定后缀（如.py）结尾的文件。通过os.walk遍历文件系统，筛选出符合条件的文件并将其路径添加到结果列表中。

使用python3.6及以上版本，通过遍历查找出某个文件夹内所有的子文件和指定后缀的所有文件，示例如下：

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os


def get_files(path='E:\\xx', rule=".py"):
    all = []
    # os.walk是获取所有的目录
    for fpath, dirs, fs in os.walk(path):
        for f in fs:
            filename = os.path.join(fpath, f)
            # 判断是否以"rule"结尾，自定义规则
            if filename.endswith(rule):
                all.append(filename)
    return all


if __name__ == "__main__":
    b = get_files(r"E:\xx\test")
    for i in b:
        print(i)