※ 引述《kilfu0701 ((￣ー+￣)キラリ)》之铭言： : ※ 引述《kadodo (想办法突破)》之铭言： : : #!/usr/bin/env python3 : : import os, time : : root = "/home/myhome" : : def get_dirs_parent(parent_dir): : : l1_all = [] : : des_dir = os.listdir(parent_dir) : : for i in range(len(des_dir)): : : if (os.path.isdir(parent_dir+"/"+des_dir[i]) & (str(des_dir[i]) != "CVS")): : : l1_all.append(parent_dir+"/"+des_dir[i]) : : #l1_all.append(des_dir[i]) : : return(l1_all) : : def get_dirs_hier2(parent_dir): : : l1_all = get_dirs_parent(parent_dir) : : l2_all = [] : : for i in range(len(l1_all)): : : l2_each = get_dirs_parent(l1_all[i]) : : for j in range(len(l2_each)): : : l2_all.append(l2_each[j]) : : return(l2_all) : : def get_dirs_hier3(parent_dir): : : l2_all = get_dirs_hier2(parent_dir) : : l3_all = [] : : for i in range(len(l2_all)): : : l3_each = get_dirs_parent(l2_all[i]) : : for j in range(len(l3_each)): : : l3_all.append(l3_each[j]) : : return(l3_all) : : dir_all = get_dirs_hier3(root) : : for i in range(len(dir_all)): : : print("tc=", i, dir_all[i]) : : 以上是我想抓出所有在第几层以下属於dir的 我知道os.walk可以用 但我想另外写一个 : : 问题来了!! 我def了三个 get_dirs_parent get_dirs_hier2 get_dirs_hier3. : : 但我觉得有可能还会有更多 hier4, hier5.... : : 请问我可不可以写成类似传一个参数进去的方式 例如 : : get_dir_hier(root, 3) 就类似 get_dirs_hier3 : : get_dir_hier(root, 2) 就类似 get_dirs_hier2 : : ... : : 请问该如何改写CODE? thanks. : 我是用recursive加上depth去纪录跑到哪一层 : 以下是范例，可能写的不好 XD : def getDir(path, depth, current=0, tmp=[]): : if depth <= current: : return tmp : dirs = os.listdir(path) : for i in range(len(dirs)): : if (os.path.isdir(path+"/"+dirs[i])): : tmp.append( [path+"/"+dirs[i], current] ) : getDir(path+"/"+dirs[i], depth, current+1) : return tmp : all = getDir('D:/dev', 3) : for i in range(len(all)): : print("tc=", i, all[i]) : 在current还没达到depth的深度时，会继续往该目录下一层去抓 : 大致上做法应该是这样，有错的地方再请更正 :) def getDir(path, depth, current=0, tmp=[]): if current >= depth: return tmp # Return a list containing the names of the entries in the directory dirs = os.listdir(path) for dir in dirs: path2 = path + "/" + dir # Return true if the pathname refers to an existing directory if os.path.isdir(path2): tmp.append((current+1, path2)) getDir(path2, depth, current+1) return tmp 很好奇 tmp 为什麽可以这样使用 tmp 不是一个 default=[] 的 List 吗? --

※ 发信站: 批踢踢实业坊(ptt.cc) ◆ From: 111.240.230.79