[python] 在指定目录下找文件

import os

# 查找当前目录下所有包含关键字的文件
def findFile(path, filekw):
    return[os.path.join(path,x) for x in os.listdir(path) if os.path.isfile(x) and os.path.split(x)[1].find(filekw)>-1]

# 获取指定目录下的次级目录
def findDir(path1):
    return[os.path.join(path1,x) for x in os.listdir(path1) if os.path.isdir(x)]

# 遍历所有子目录文件
def listAll(p, k):
    for a in findFile(p,k):
        print(a)
    for b in findDir(p):
        listAll(b,k)

if __name__=="__main__":
    listAll('.','py')