指定位置下搜索文件（Python）

import os
def file_search(str_search):
    dir = os.getcwd()#得到当前目录
    for root, dirs, files in os.walk(dir):#遍历根目录
        for file in files:
            if str_search in file:
                path = os.path.join(root,file)#合并路径
                relpath = os.path.relpath(path,dir)#相对路               径，去掉当前位置
                print(relpath)


if __name__ == '__main__':
    str_search = input('输入字符串，将打印出包含该字符串的文件的相对路径:')
    file_search(str_search)

在os.walk中root保存路径，
for file in files，只检索文件名字，不检索dirs（文件夹）