python遍历文件夹取到文件并输出每种文件有几个

1. 遍历某个文件夹下的所有文件，并统计各个文件类型的个数。
import os#引入os模块

def lister(curr_dir):
stack=[curr_dir]
x = 0#定义mp4文件
y = 0#定义txt文件
while stack:
dir=stack.pop()
for path in os.listdir(dir):
file_path=os.path.join(dir,path)
if os.path.isdir(file_path):
stack.append(file_path)
else:
print(path)#输出每个文件
a=path.split(".")#取出后缀名
for i in a:#遍历取到后缀名
if i==‘mp4’:#得到mp4文件个数
x+=1
elif i==“txt”:#得到txt文件个数
y+=1
print(‘其中有视频%s部，小说%s本’ % (x, y))
if name==‘main’:
lister(‘data’)