python 获取当前目录\指定目录下的所有指定格式的文件

最新推荐文章于 2024-12-15 09:02:10 发布

原创最新推荐文章于 2024-12-15 09:02:10 发布 · 1.5k 阅读

7 ·

CC 4.0 BY-SA版权

python 自建函数专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种Python方法，用于获取指定目录下特定格式的文件列表。通过使用os模块，可以遍历目录并筛选出符合指定后缀的文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os
#获取当前目录下的指定格式文件
def getFiles(suffix):                   #suffix为str，示例:   '.txt'      '.json'       '.h5'
	BASE_DIR = os.path.dirname(os.path.abspath(__file__))       #这里如果有报错，则在__file__上加引号
	res = []
	for root,directory,files in os.walk(BASE_DIR):
    	for filename in files:
        	name,suf = os.path.splitext(filename)
        	if suf == suffix:
           		res.append(os.path.join(root,filename))
	return(res)    
#获取指定目录下的指定格式的文件
dir = '/usr/file'
def getFiles(dir,suffix):
	res = []
	for root,directory,files in os.walk(dir):
    	for filename in files:
        	name,suf = os.path.splitext(filename)
        	if suf == suffix:
           		res.append(os.path.join(root,filename))
	return(res)