python获取一个目录下所有jpg文件

输入：文件夹目录名称
输出：以列表形式返回该目录下包括其子目录下所有的后缀名为.jpg的文件名

import os
def get_fileNames(rootdir):
fs = []
for root, dirs, files in os.walk(rootdir,topdown = True):
for name in files:
_, ending = os.path.splitext(name)
if ending == ".jpg":
fs.append(os.path.join(root,name))
return fs