pyton2路径递归问题
想用Python2做这样一个函数。给定一个路径sourcedir,给定一个子路径级别数N,自动搜索出sourcedir下面到N层子路径的所有子路径。例如下列路径结构
/a/b1/c1
/a/b2/c2
/a/b3/c3/d3
/a/b4/c4/d4/e4
/a/b5/c5/d5/e5/f5
sourcedir="/a"
当N=1时,函数返回["b1", "b2", "b3", "b4", "b5"]
当N=2时,函数返回["b1/c1", "b2/c2", "b3/c3", "b4/c4", "b5/c5"]
当N=3时,函数返回["b3/c3/d3", "b4/d4/c4", "b5/c5/d5"]
当N=4时,函数返回["b4/c4/d4/e4", "b5/c5/d5/e5"]
当N=5时,函数返回["b5/c5/d5/e5/f5"]
当N>=6时,函数返回[ ]
想了很久都没有实现。有大神可以帮忙实现么?
多谢了
我原来的思路是先遍历sourcedir下所有的子路径,然后再从里面选择需要层数的子路径返回。但是当sourcedir下面有很多子路径和文件时。这种方法慢的要死。需要一种递归方法,只递归N级子路径就停止才行。这个方法想了很久都没有做出来。需要高手帮忙。