python读取一个文件夹下指定文件类型的所有文件名

思路

使用os类的walk方法（import os），获取文件夹下所有根目录roots、文件夹dirs和文件files。这里只要文件就行，遍历文件夹下所有的文件，确定指定类型的文件添加到列表。
有两种方法确定：

1. 判断扩展名是否在文件路径中
2. os.path.splitext()分离扩展名，判断是否与指定类型相同

代码

方法一

def get_filename(path,filetype):  # 输入路径、文件类型 例如'.csv'
    name = []
    for root,dirs,files in os.walk(path):
        for i in files:
            if filetype+' ' in i+' ':    # 这里后面不加一个字母可能会出问题，加上一个（不一定是空格）可以解决99.99%的情况
                name.append(i)    
    return name            # 输出由有后缀的文件名组成的列表

方法二

def get_filename(path,filetype):  # 输入路径、文件类型例如'.csv'
    name = []
    for root,dirs,files in os.walk(path):
        for i in files:
            if os.path.splitext(i)[1]==filetype:
                name.append(i)    
    return name            # 输出由有后缀的文件名组成的列表