python指定目录中查找数字开头文件

import re
from os import walk
from os import path

input_dir=input("please input the seach dir:")  #输入搜索目录

for parent,dirNames,fileNames in walk(input_dir):  #根据os.walk输出目录信息
    for fileName in fileNames:                     #遍历所有文件名
        r='^[1-9]+.py'                             
        res=re.search(r,fileName)                  #查找指定正则的文件名
        if res:
            print (path.join(parent,fileName))     #拼接目录和文件名，输出