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

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

思路

代码

思路

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

有两种方法确定：

判断扩展名是否在文件路径中

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 # 输出由有后缀的文件名组成的列表