python 遍历目录或文件

python 遍历目录或文件

如果想得到某个目录下面的所有文件名,用 python 怎么做?

先看代码:

import os,sys

def explore(dir):
    for root, dirs, files in os.walk(dir):
        print('debug: ', root, dirs, files) # 这行用来调试,帮助理解代码    
        for file in files:
            path = os.path.join(root, file)
            print(path)

 
def main():
    for path in sys.argv[1:]:           
        if os.path.isdir(path):
            explore(path)
			
			
if __name__ == "__main__":
    main()

假设上面的代码保存为文件 get_dirs.py

我们实操一下。

当前目录结构如下:

$ tree
.
├── a
│   └── a1
│       └── a11 // 目录
├── a.txt 
├── b   // 目录
├── b.txt
├── c.txt
├── get_dirs.py
└── ss
    └── s.txt

结果是:

$ python3 get_dirs.py ./
debug:  ./ ['a', 'b', 'ss'] ['a.txt', 'b.txt', 'c.txt', 'get_dirs.py']
./a.txt
./b.txt
./c.txt
./get_dirs.py
debug:  ./a ['a1'] []
debug:  ./a/a1 ['a11'] []
debug:  ./a/a1/a11 [] []
debug:  ./b [] []
debug:  ./ss [] ['s.txt']
./ss/s.txt

代码解读如下。

    for root, dirs, files in os.walk(dir):
        print('debug: ', root, dirs, files) # 这行用来调试,帮助理解代码    
        for file in files:
            path = os.path.join(root, file)
            print(path)

第 1 行的 os.walk 的函数声明为:

walk(top, topdown=True, onerror=None, followlinks=False)

参数:

  • top 是你所要遍历的目录的地址
  • topdown 为真,则优先遍历 top 目录,否则优先遍历 top 的子目录(默认为真)
  • onerror 需要一个 callable 对象,当 walk 需要异常时,会调用
  • followlinks 如果为真,则会遍历目录下的快捷方式(linux 下是 symbolic link)实际所指的目录(默认为假)

os.walk 的返回值是一个生成器(generator),也就是说我们需要不断地遍历它,以获得所有内容。

可以理解为 os.walk 会遍历所有的目录(不包括文件),每次遍历都会返回一个三元组:(root,dirs,files)。

  • root 表示当前正在遍历的目录
  • dirs 是一个 list ,会列出 root 下所有的目录(不包括子目录)
  • files 也是一个 list , 会列出 root 下所有的文件(不包括子目录里的文件)

知道了这些,你再对比上面的打印结果就会豁然开朗。

代码第 4 行,path = os.path.join(root, file)

这句是什么意思呢?其实就是把 root 和 file 拼接到一起,组成完整的文件名。

举个例子就明白了。

>>> import os
>>> os.getcwd()
'/mnt/hgfs/vm_share/grep_test'
>>> my_path = os.path.join(os.getcwd(), "hello")
>>> print(my_path)
/mnt/hgfs/vm_share/grep_test/hello

参考资料

【1】python中os.walk的用法

【2】os.path.join()用法

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值