python-walk遍历目录及查找目录小例子

1、walk遍历目录

import  os

for cur_dir,dirs,files in os.walk('.'): #遍历目录下的所有文件及文件夹
    print('当前目录%s'%cur_dir)
    print('文件夹%s'%dirs)
    print('文件%s'%files)


/Library/Frameworks/Python.framework/Versions/3.7/bin/python3.7 /Users/dongyf/Documents/python/besttest_study/test.py
当前目录.
文件夹['cnz', '.idea']
文件['day2.py', 'xx-20190610.log', 'day3_randompasswd.py', 'control.py', 'redom_test.py', 'test.py', 'c.txt', 'dyf.py', 'a.txt', 'classtime.py', 'study_day1.py', 'c.json', 'access.log', 'day3_goodsmanage.py']
当前目录./cnz
文件夹['ycc']
文件['b.py']
当前目录./cnz/ycc
文件夹[]
文件['a.py']
当前目录./.idea
文件夹[]
文件['besttest_study.iml', 'workspace.xml', 'modules.xml', 'misc.xml']

2、例子：

'''
# 2、写一个查找文件内容的程序
def find_content(path, key_word):
    pass
在某个目录下，找到文件内容里面包含这个关键字的文件名。
'''
import  os
#找目录下含该关键字的文件名及其路径
def find_content(path, key_word):
    filename = []
    for cur_dir, dirs, files in os.walk(path): #遍历目录文件名
        for name in files:
            os.chdir(cur_dir) #切换到当前路径
            r=readfile(name) #读文件内容
            if key_word in r: #文件内容是否包含关键字
                filename.append(name + "   " + cur_dir) #文件名及路径添加到数组
    return filename  #返回数组

#读文件内容函数
def readfile(name):
    f = open(name, encoding='utf-8') #读模式打开文件
    r= f.read() #读文件
    f.close()  #关闭文件
    return  r #返回文件内容

key_word='123' #定义关键字
path=r'C:\zwj\besttest\logs' #文件目录
print(find_content(path,key_word)) #调取找文件关键字函数