import os
def search(dststr, path=['.']):
for n in path:
dirlist = [x for x in os.listdir(n) if os.path.isdir(os.path.join(n, x))]
filelist = [x for x in os.listdir(n) if os.path.isfile(os.path.join(n, x)) and dststr in os.path.splitext(x)[0]]
if len(filelist) > 0:
print(n, ':')
for m in filelist:
print(m)
print('\n')
path = [os.path.join(n, x) for x in dirlist]
search(dststr, path)
return 'Done'
dststr = '2'
search(dststr)
梦想家高先森