import os
pth = "/path/to/notes"
def getCats(pth):
cats = []
subcats = []
for (dirpath, dirnames, filenames) in os.walk(pth):
#print dirpath+"\n\t", "\n\t".join(dirnames), "\n%d files"%(len(filenames))
if dirpath == pth:
cats = dirnames
else:
subcats.extend(dirnames)
dirnames[:]=[] # don't walk any further downwards
# subcats = list(set(subcats)) # uncomment this if you want 'subcats' to be unique
return (cats, subcats)
你可以利用^{} 。
简单的两级遍历:
os.walk非常适合这个。默认情况下,它将执行自上而下的遍历,您可以通过将“dirnames”设置为空,在第二级轻松终止它。
相关问题 更多 >
编程相关推荐