利用Python查看目录中的文件示例详解
前言
我们在日常开发中，经常会遇到一些关于文件的操作，例如，实现查看目录内容的功能。类似Linux下的tree命令。统计目录下指定后缀文件的行数。
功能是将目录下所有的文件路径存入list中。可以加入后缀判断功能，搜索指定的后缀名文件。主要利用递归的方法来检索文件。
仿造tree功能示例代码
Python2.7
列出目录下所有文件
递归法
importos deftree_dir(path,c_path='',is_root=True): """ Getfilelistunderpath.Like'tree' :parampathRootdir :paramc_pathChilddir :paramis_rootCurrentisrootdir """ res=[] ifnotos.path.exists(path): returnres forfinos.listdir(path): ifos.path.isfile(os.path.join(path,f)): ifis_root: res.append(f) else: res.append(os.path.join(c_path,f)) else: res.extend(tree_dir(os.path.join(path,f),f,is_root=False)) returnres
下面是加入后缀判断的方法。在找到文件后，判断一下是否符合后缀要求。不符合要求的文件就跳过。
deftree_dir_sur(path,c_path='',is_root=True,suffix=''):
"""Getfilelistunderpath.Like'tree'
:parampathRootdir
:paramc_pathChilddir
:paramis_rootCurrentisrootdir
:paramsuffixSuffixoffile
"""
res=[]
ifnotos.path.exists(path)ornotos.path.isdir(path):
returnres
forfinos.listdir(path):
ifos.path.isfile(os.path.join(path,f))andstr(f).endswith(suffix):
ifis_root:
res.append(f)
else:
res.append(os.path.join(c_path,f))
else:
res.extend(tree_dir_sur(os.path.join(path,f),f,is_root=False,suffix=suffix))
returnres
if__name__=="__main__":
forpintree_dir_sur(os.path.join('E:\ws','rnote','Python_note'),suffix='md'):
printp
统计目录下指定后缀文件的行数
仅适用os中的方法，仅检索目录中固定位置的文件
#-*-coding:utf-8-*-
importos
defcount_by_categories(path):
"""Findalltargetfilesandcountthelines"""
ifnotos.path.exists(path):
return
c_l_dict=dict()#e.g.{category:lines}
category_list=[cateforcateinos.listdir(path)if
os.path.isdir(os.path.join(path,cate))andnotcate.startswith('.')]
forcategory_dirincategory_list:
line_count=_sum_total_line(os.path.join(path,category_dir),'.md')
ifline_count>0:
c_l_dict[category_dir]=line_count
returnc_l_dict
def_sum_total_line(path,endswith='.md'):
"""Getthetotallinesoftargetfiles"""
ifnotos.path.exists(path)ornotos.path.isdir(path):
return0
total_lines=0
forfinos.listdir(path):
iff.endswith(endswith):
withopen(os.path.join(path,f))ascur_f:
total_lines+=len(cur_f.readlines())
returntotal_lines
if__name__=='__main__':
note_dir='E:/ws/rnote'
ca_l_dict=count_by_categories(note_dir)
all_lines=0
forkinca_l_dict.keys():
all_lines+=ca_l_dict[k]
print'alllines:',str(all_lines)
printca_l_dict
以笔记文件夹为例，分别统计分类目录下文件的总行数，测试输出
alllines:25433
{'flash_compile_git_note':334,'Linux_note':387,'Algorithm_note':3637,'Comprehensive':216,'advice':137,'Java_note':3013,'Android_note':11552,'DesignPattern':2646,'Python_note':787,'kotlin':184,'cpp_note':279,'PyQt_note':439,'reading':686,'backend':1136}
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流，谢谢大家对毛票票的支持。