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