Python遍历目录下的文件(os.walk 、os.listdir的用法)

Python遍历目录下的文件(os.walk 、os.listdir的用法)

最近想从一个目录转移一些文件到另外一个目录,因为文件有指定的后缀,所以想着用python来转移,不做cv工程师,那就涉及到如何遍历目录(子目录)?

下面主要介绍两种方式,os.walk() 与 os.listdir() ,使用的时候根据下面的需求查看使用

1、os.walk()

定义:

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

说明:

返回:

os.walk 的返回值是一个可遍历的对象,每次遍历返回的是一个三元元组,可用root,dirs,files接收

需求:

​ 现在有一个test目录,我想把test下面的(包含其子目录)所有txt文件移动到一个文件去

# tree (test目录的情况如下)
├─test
	├─A
        ├─r
            |─r_1.txt
            |─r_2.txt
        ├─t
        	|─t_1.txt
        	|─t_1.txt
    ├─B
        ├─r
            |─b_r_1.txt
            |─b_r_2.txt
        ├─t
        	|─b_t_1.txt
        	|─b_t_1.txt
    ├─C
        ├─r
            |─c_r_1.txt
            |─c_r_2.txt
        ├─t
        	|─c_t_1.txt
        	|─c_t_1.txt

代码:

import os

path = "C:\\Users\\Administrator\\Desktop\\test"

for root, dirs, files in os.walk(path):
    print("root:", root)
    print("dirs:", dirs)
    print("files", files)

可以看到输出:(对照着上面的字段说明)

root: C:\Users\Administrator\Desktop\test
dirs: ['A', 'B', 'C']
files []

root: C:\Users\Administrator\Desktop\test\A
dirs: ['r', 't']
files []

root: C:\Users\Administrator\Desktop\test\A\r
dirs: []
files ['r_1.txt', 'r_2.txt']

root: C:\Users\Administrator\Desktop\test\A\t
dirs: []
files ['t_1.txt', 't_2.txt']

root: C:\Users\Administrator\Desktop\test\B
dirs: ['r', 't']
files []

root: C:\Users\Administrator\Desktop\test\B\r
dirs: []
files ['b_r_1.txt', 'b_r_2.txt']

root: C:\Users\Administrator\Desktop\test\B\t
dirs: []
files ['b_t_1.txt', 'b_t_2.txt']

root: C:\Users\Administrator\Desktop\test\C
dirs: ['r', 't']
files []

root: C:\Users\Administrator\Desktop\test\C\r
dirs: []
files ['c_r_1.txt', 'c_r_2.txt']

root: C:\Users\Administrator\Desktop\test\C\t
dirs: []
files ['c_t_1.txt', 'c_t_2.txt']

2、os.listdir()

定义:

os.listdir(path)

说明:

返回:

os.listdir的返回值是一个列表,列表里面存储该path下面的子目录的名称

代码:

import os

path = "C:\\Users\\Administrator\\Desktop\\test"

for dir in os.listdir(path):
    print(dir)

输出:

A
B
C

如果想得到最后的文件,可以通过os.path.join来连接字符串,然后不停的遍历,获取到最后的文件

import os

path = "C:\\Users\\Administrator\\Desktop\\test"

for dir in os.listdir(path):
    # child_dir : A B C
    child_dir = os.path.join(path, dir)
    for d in os.listdir(child_dir):
        # 目录 : r  t
        ddd= os.path.join(path, dir ,d)
        for file in os.listdir(ddd):
            # 文件名称
            print(file)

总结

个人觉得os.walk会好用,简单很多,代码编写量也少

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年3月10日
下一篇 2023年3月10日

相关推荐