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

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

你可能感兴趣的:(python)