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

import osdef 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（文件夹）