使用os类的walk方法(import os),获取文件夹下所有根目录roots、文件夹dirs和文件files。这里只要文件就行,遍历文件夹下所有的文件,确定指定类型的文件添加到列表。
有两种方法确定:
方法一
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 # 输出由有后缀的文件名组成的列表