python获取文件夹下的所有文件名

os模块下有两个函数：os.walk(),os.listdir()

os.walk()：

# -*- coding: utf-8 -*-
    import os
    def getFile(dirPath):
        for root, dirs, files in os.walk(file_dir):
        	for file in files:
        	#os.path.splitext()函数将路径拆分为文件名+扩展名
            	if os.path.splitext(file)[1] == '.jpg':
            		print("文件名后缀为jpg")
            print(root) #当前目录路径
            print(dirs) #当前路径下所有子目录
            print(files) #当前路径下所有非目录子文件

os.listdir():

# -*- coding: utf-8 -*-
    import os
    def listdir(path, list_name):  #传入存储的list
        for file in os.listdir(path):
            file_path = os.path.join(path, file)
            if os.path.isdir(file_path):
                listdir(file_path, list_name)
            else:
                list_name.append(file_path)