在Python开发中,将常用功能封装成为接口,并放入Utils工具类中,直接调用,可以提升效率。
常用的函数有:
文件夹遍历
文件夹创建
文件读取
时间可读
时间统计
安全除法
双列表排序
配置读取
脚本路径
Numpy判空
文件夹遍历
遍历文件夹函数提供的功能和扩展,如下:
返回文件的路径和名称;
根据后缀名筛选文件;
去除隐藏文件,即以“.”开头的文件;
实现:
def traverse_dir_files(root_dir, ext=None):
"""
列出文件夹中的文件, 深度遍历
:param root_dir: 根目录
:param ext: 后缀名
:return: [文件路径列表, 文件名称列表]
"""
names_list = []
paths_list = []
for parent, _, fileNames in os.walk(root_dir):
for name in fileNames:
if name.startswith('.'): # 去除隐藏文件
continue
if ext: # 根据后缀名搜索
if name.endswith(tuple(ext)):
names_list.append(name)
paths_list.append(os.path.join(parent, name))
else:
names_list.append(name)
paths_list.append(os.path.join(parent, name))
paths_list, names_list = sort_two_list(paths_list, names_list)
return paths_list, names_list
复制代码 |