Re: [問題] 請問該怎麼改寫這段CODE

看板Python作者kilfu0701 ((￣ー+￣)キラリ)時間13年前 (2012/03/14 12:24)推噓0(0推 0噓 0→)

留言0則, 0人參與討論串2/4 (看更多)

※ 引述《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的深度時，會繼續往該目錄下一層去抓大致上做法應該是這樣，有錯的地方再請更正 :) -- ※ 發信站: 批踢踢實業坊(ptt.cc) ◆ From: 203.74.124.193