linux下用python统计某个目录下指定后缀文件的个数

#!/usr/bin/python

#filename:traversefileInPython

import os
import os.path
dir="/home/test"

m=0

for parentdir,dirname,filenames in os.walk(dir):

for filename in filenames:

print filename

if os.path.splitext(filename)[1]=='.wav':
m=m+1
print m