前言
我們在日常開發中，經常會遇到一些關于文件的操作，例如，實現查看目錄內容的功能。類似Linux下的tree命令。統計目錄下指定后綴文件的行數。
功能是將目錄下所有的文件路徑存入list中。可以加入后綴判斷功能，搜索指定的后綴名文件。主要利用遞歸的方法來檢索文件。
仿造 tree 功能示例代碼
Python2.7
列出目錄下所有文件
遞歸法
import osdef tree_dir(path, c_path='', is_root=True): """ Get file list under path. Like 'tree' :param path Root dir :param c_path Child dir :param is_root Current is root dir """ res = [] if not os.path.exists(path): return res for f in os.listdir(path): if os.path.isfile(os.path.join(path, f)): if is_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)) return res
下面是加入后綴判斷的方法。在找到文件后，判斷一下是否符合后綴要求。不符合要求的文件就跳過。
def tree_dir_sur(path, c_path='', is_root=True, suffix=''): """ Get file list under path. Like 'tree' :param path Root dir :param c_path Child dir :param is_root Current is root dir :param suffix Suffix of file """ res = [] if not os.path.exists(path) or not os.path.isdir(path): return res for f in os.listdir(path): if os.path.isfile(os.path.join(path, f)) and str(f).endswith(suffix): if is_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)) return resif __name__ == "__main__": for p in tree_dir_sur(os.path.join('E:/ws', 'rnote', 'Python_note'), suffix='md'): print p統計目錄下指定后綴文件的行數
僅適用os中的方法，僅檢索目錄中固定位置的文件
# -*- coding: utf-8 -*-import osdef count_by_categories(path): """ Find all target files and count the lines """ if not os.path.exists(path): return c_l_dict = dict() # e.g. {category: lines} category_list = [cate for cate in os.listdir(path) if os.path.isdir(os.path.join(path, cate)) and not cate.startswith('.')] for category_dir in category_list: line_count = _sum_total_line(os.path.join(path, category_dir), '.md') if line_count > 0: c_l_dict[category_dir] = line_count return c_l_dictdef _sum_total_line(path, endswith='.md'): """ Get the total lines of target files """ if not os.path.exists(path) or not os.path.isdir(path): return 0 total_lines = 0 for f in os.listdir(path): if f.endswith(endswith): with open(os.path.join(path, f)) as cur_f: total_lines += len(cur_f.readlines()) return total_linesif __name__ == '__main__': note_dir = 'E:/ws/rnote' ca_l_dict = count_by_categories(note_dir) all_lines = 0 for k in ca_l_dict.keys(): all_lines += ca_l_dict[k] print 'all lines:', str(all_lines) print ca_l_dict以筆記文件夾為例，分別統計分類目錄下文件的總行數，測試輸出
all lines: 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}總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作能帶來一定的幫助，如果有疑問大家可以留言交流，謝謝大家對VEVB武林網的支持。
新聞熱點
疑難解答
