python遍历目录就是这么简单

Posted 2023-03-13

参考技术A

有时我们有列出目录下都有哪些文件和子目录的需求，这种情况是有现成命令可用的，比如windows下的dir命令，linux下的ls命令都可以，那我们用python代码怎么实现呢？

我们利用python丰富的库很容易就能实现一个简易版本，下面我们就用4种方法来实现它。

一、使用os.popen

os.popen工作原理是新建一个子进程，然后用这个子进程执行命令，父进程与子进程间通过管道进行通信。

根据调用popen时的传参，我们可以通过管道读取子进程的输出也可以向子进程写数据，默认是读取子进程的输出。

从以上描述可以看出popen是非常通用的，不是只能用于我们这个例子哦。

那我们开始用它实现我们的需求吧，代码如下：

哈哈，是不是很简单，这种方式虽然能达到目的但其实并不是我们想要的，我们本来就是要实现ls的，结果调用了ls，所以严格意义上来说我们并没有实现ls，那让我们继续往下看其它方法吧，嘿嘿。

二、使用glob.glob

glob可以根据你使用的通配符对文件进行匹配，利用这个特性我们可以列出当前目录下都有哪些文件和子目录，如下代码：

三、使用os.listdir

os.listdir同样可以列出某个目录下都有哪些文件和子目录，如下代码：

四、使用os.walk

os.walk在遍历目录方面非常强大，它不但可以遍历你需要的目录，也可以递归遍历子目录且递归的深度可以用代码控制，下面让我们分别看下怎么遍历整个目录树以及怎么控制深度吧。

os.walk默认是遍历整个目录树的，如下代码就会递归打印出当前目录下所有文件：

那我们怎么控制遍历的深度，比如只遍历n层呢？其实很简单，只需要定义一个深度变量，然后到达n后跳出循环即可，如下代码就只遍历1层：

至此我们已经写完4种方法了，如果你还有其他方法，欢迎评论交流。