Python初级第3课——我的Tree命令（只读课堂）

欢迎来到“只读课堂”！

这节课我们是专门针对于上一节课的os模块来制作一个Tree命令，如果你没有完成上一节课的课程，就来学这一堂课，那么你将会很吃力。

In [1]:

#tree命令就是返回一个目录下所有的文件名。

下面，我们来正式开始敲代码

In [2]:

import os

In [3]:

#递归

In [4]:

#目录->当前下面的 文件，文件夹->文件，文件夹

In [ ]:

dirpath = input('请输入你要遍历的目录') def getdir(dirpath): #当路径为假 if not dirpath: dirpath = os.getcwd() #默认情况下以当前路径开始 #有了路径之后 mylist = os.listdir(dirpath) #取出来所有的文件和文件夹 #判断哪个是文件夹 for name in mylist: print(name) #这里可以避免相对路径，手动来拼接，变成绝对路径 name = os.path.join(dirpath,name) if os.path.isdir(dirpath+'\\'+name): getdir(name) getdir(dirpath)

In [ ]:

#将以上代码输入编译器里，就可以使用了。

美化

In [ ]:

#然后我们看到打印出来的东西是不是不好看啊，排列很乱，所以我们加几个东西：

In [ ]:

import os dirpath = input('请输入你要遍历的目录') def getdir(dirpath): if not dirpath: dirpath = os.getcwd() mylist = os.listdir(dirpath) for name in mylist: print('-'*level + '|' + name) name = os.path.join(dirpath,name) if os.path.isdir(dirpath+'\\'+name): getdir(name,level) getdir(dirpath)

最终代码：

In [ ]:

import os dirpath = input('请输入你要遍历的目录') def getdir(dirpath): if not dirpath: dirpath = os.getcwd() mylist = os.listdir(dirpath) for name in mylist: print('-'*level + '|' + name) name = os.path.join(dirpath,name) if os.path.isdir(dirpath+'\\'+name): getdir(name,level) getdir(dirpath)

以上就是我们这节课“我的Tree命令”的源码，它还是有可以改进的地方的

本次只读课堂的python教程就到这了，欢迎下一次的收看！