51工具盒子

依楼听风雨
笑看云卷云舒,淡观潮起潮落

python - 遍历目录

一门语言,如果知道了基本语法,知道输入输出、文件操作、数据库的操作、通信(socket等),基本上就可以开始开发了。剩下的知识就是在开发中慢慢学习与积累。这也是我学习一门新语言首先去了解的部分。现在的语言五花八门,其实归根结底也就是这些内容。

大家也可能在学习中感到迷茫,好像新的东西自己不知道怎么去用。所以,把上面这些点都能看到,就开始做自己头脑中想过的那些程序吧。

说了这么多废话,进入主题......我曾经写了C语言的遍历目录PHP的遍历目录,今天来说一下python遍历目录,也算给大家一个对比,以及学习我上面说的:文件操作。


程序要实现的目的是遍历某个目录,并把其中的所有文件中某段内容替换掉。

import os

def replace_str(FilePath,SourceStr,ObjectStr):
    file = open(FilePath,"r")
    str = file.read()
    str = str.replace(SourceStr,ObjectStr)
    file.close()
    file = open(FilePath,"w")
    file.write(str)
    file.close()

def getfile(path):
    for file in os.listdir(path):
        file = os.path.join(path,file)
        if os.path.isdir(file):
            getfile(file)
        else:
            replace_str(file,"abcd","")

rootpath = "e:\\01"
getfile(rootpath)

我们要遍历目录,首先导入os模块。import os

先跳过两个函数,看到我们最下面两行。我们就是需要遍历rootpath目录,将之传入getfile函数。

getfile是个递归函数,遍历目录是一定要用递归的。运用os.listdir()获取所有的文件(包括目录和文件),获取到的是一个list,用for ..in遍历list,然后用os.path.isdir()函数判断当前的文件是否是一个目录。

如果是目录则递归执行getfile()函数,否则就进入replace_str()函数。

大概就是这个过程。代码也很简单,结合我前两篇文章,应该不难理解。主要是对比,看看三种语言完成一个任务,各自有什么优劣。



补记:


遍历目录,其实python的os模块已经给出了一个函数。

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

它返回的对象是一个generator类型对象,我们要使用for语句来遍历它:


generator = os.walk(r"E:\Python\pro")
for i in generator:
    print i

我们会发现,i 实际上是一个三元tuple,分别是此时的目录,此目录下的文件夹,此目录下的文件。所以我们上面的getfile函数可以简化如下:

def getfile(path):
    generator = os.walk(path)
    for (now_dir, _, file_list) in generator:
        for file in file_list:
            file = os.path.join(now_dir, file)
            replace_str(file, "abc", "")


赞(2)
未经允许不得转载:工具盒子 » python - 遍历目录