Python 获取文件下所有文件类型名称、文件夹不存在时自动创建

返回文件夹下所有指定文件名

有时我们需要统计一下当前文件夹下中包含全部的 png 文件或者说含有 png 图片数量，此功能常用于文件检索； 标准库 os 虽然有一些很强大的函数，但没有一个能满足我们这个需求，那么我想下面的这个代码块或许能帮到你！

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

文件夹不存在时自动创建

这个功能在日常开发办公中会经常用到，主要用到了 os 模块的两个函数

• os.path.exists(path) 判断文件夹是否存在；
• os.makedirs(path) 创建文件夹

import os  # Import the OS module

MESSAGE = 'The directory already exists.'
TESTDIR = 'testdir'
try:
    home = os.path.expanduser("~")  
    print(home)  # Print the location

    if not os.path.exists(os.path.join(home, TESTDIR)):  # os.path.join() for making a full path safely
        os.makedirs(os.path.join(home, TESTDIR))  # If not create the directory, inside their home directory
    else:
        print(MESSAGE)
except Exception as e:
    print(e)