使用python3.x实现文件遍历

使用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)