【Python】读取文件夹中指定类型的文件名

在网上看了很多人写的关于读取文件夹中的指定类型的文件名的文章，感觉集中于读取某一指定类型的文件名，而在实际过程中可能会遇到需要读取不同格式的文件，因而自己创建了相关的函数，希望能够给大家带来帮助。

从文件夹中读取所有单一格式的文件名

import os
def read_file(path,file_type): #path 是指需要提取的文件夹所在路径;file_type是指文件的拓展名;txt格式的文件则输入‘.txt’    
    filenames = os.listdir(path) 
    filenames1 = []
    for filename in filenames:
        if os.path.splitext(filename)[1] == file_type:
            filenames1.append(filename)
    return filenames1
###显示文件名####
#file_name=read_file()
#print(file_name)
########

从文件夹中读取多种格式的文件名

import os
def read_files(path,file_types):#path 是指需要提取文件的文件夹所在路径；file_types指的是需要读取的文件拓展名的集合，例如jpg，png两种图片格式的文件则输入['.jpg','.png']
    filenames = os.listdir(path)
    filenames1=[]
    for file_type in file_types:
        for filename in filenames:
            if os.path.splitext(filename)[1] == file_type:
                filenames1.append(filename)
    return filenames1
#file_name=read_files()
#print(file_name)
########