欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python遍历读取文件下的所有指定类型的文件

程序员文章站 2022-05-28 11:46:57
...

python遍历读取文件下的所有指定类型的文件

2018年08月12日 09:53:23 Blue__Bubble 阅读数:1075

#!/usr/bin/env python3

-- coding:utf-8 --

1. for-in dir/subdir to get the filesname

2. splitext filename to filter

import os

def getFiles(dir, suffix): # 查找根目录,文件后缀 
    res = []
    for root, directory, files in os.walk(dir):  # =>当前根,根下目录,目录下的文件
        for filename in files:
            name, suf = os.path.splitext(filename) # =>文件名,文件后缀
            if suf == suffix:
                res.append(os.path.join(root, filename)) # =>吧一串字符串组合成路径
    return res

for file in getFiles("./", '.py'):  # =>查找以.py结尾的文件
    print(file)