返回文件夹下所有指定文件名
有时我们需要统计一下当前文件夹下中包含全部的 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)
