一文搞定Python操做文件和目录

本文首发于公众号“AntDream”，欢迎微信搜索“AntDream”或扫描文章底部二维码关注，和我一块儿天天进步一点点

文件和目录操做是很常见的功能，这里作个简单的总结，包括注意事项和实际的实现代码，基本平常开发都够用了

目录操做

判断目录或是文件是否存在

os.path.exists(path)
复制代码

判断是不是文件或是目录

# 若是文件或是目录不存在也会返回False
os.path.isfile(path)
os.path.isdir(path)
复制代码

建立/删除目录

os.mkdir(path)
os.rmdir(path)
复制代码

获得当前的目录名称

os.path.split(dir_path)[1]
复制代码

这个方法既简单又实用，好比咱们输入一个目录路径：

os.path.split('/usr/projects/project1')[1]
# 输出
project1

# 实际上，上面的代码是分了2步走：
# 第1步：利用os.path.split方法分割路径
os.path.split('/usr/projects/project1')
# 输出
('/usr/projects', 'project1')
# 而后第2步取出第2个结果，也就是当前的目录名字
复制代码

循环遍历目录

循环遍历目录有2种方式，一种是常规的递归方法，相似下面这样：

def list_file(path):
    for file in os.listdir(path):
        if os.path.isdir(os.path.join(path, file)):
            list_file(os.path.join(path, file))
        print(file)
复制代码

还有一种更好的方式就是用os.walk方法，相似下面这样：

def list_file_by_walk(dir_path):
    for home, dirs, files in os.walk(dir_path):
        print(home)
        print(dirs)
        # files中是文件列表
        for file_name in files:
            print(file_name)
复制代码

os.walk方法的几个参数简单介绍一下：

• home：这个是当前遍历的目录，好比说是 /usr/projects/
• dirs：是这个目录（home目录）下面的全部文件夹（也就是子目录）
• files：是这个目录（home目录）下全部的文件

因此从上面的参数中能够看出，os.walk方法会逐一地遍历初始目录下面的全部目录和文件

常见的文件操做

跟上面获得目录名相似的是获得文件的后缀

os.path.splitext(file_path)
复制代码

这个方法若是输入的是文件的路径，好比：

a = '/usr/projects/project1/test.txt'
os.path.splitext(a)
# 输出
('/usr/projects/project1/test', '.txt')
# 因此跟上面相似，咱们能够直接拿到文件的后缀
os.path.splitext(a)[1]
复制代码

文件的全路径

# 须要文件的目录路径和文件名
os.path.join(dir_name, file_name)
复制代码

这个方法也颇有用，由于咱们在打开一个文件的时候都须要知道文件的路径

好比在上面循环遍历目录的例子中，咱们能够这样打开编辑文件：

def list_file_by_walk(dir_path):
    for home, dirs, files in os.walk(dir_path):
        print(home)
        print(dirs)
        # files中是文件列表
        for file_name in files:
            # 这个file_name只是一个文件名
            print(file_name)
            # 若是咱们须要打开文件进行编辑读写操做，那就须要文件的路径
            # 用os.path.join方法就能够快速获得文件路径
            file_path = os.path.join(home, file_name)
复制代码

须要注意的是，不要直接用字符串相加的方式来拼接，这样子会有兼容问题，好比windows上可能路径就是错的了

读写文件

打开文件

with open(file, 'r') as f:
    for line in f.readlines():
        print line
复制代码

你们只要记住上面打开文件的格式就好了，也就是这句：

with open(file_path, 打开的模式) as f
# 咱们就拿到了文件对象f，能够对文件进行操做了，好比读写等
复制代码

缘由是文件读写是IO操做，须要及时关闭打开的文件，上述with open() as f的方式会自动帮你关闭文件的，省得本身忘记关

文件的打开模式

文件的打开模式有不少种，好比只读、只写、追加模式等等，具体能够见下图：

模式有这么多，掌握常见的几种就能够了，其余的等用到了再去了解。须要注意如下几点：

• 建立文件，只须要打开一个文件便可：

# 只要这样一句就建立了一个新文件，也就是打开之后关闭文件
with open(new_file, 'w') as f:
复制代码

• 须要特别注意的是，以w只写模式打开一个文件后会把原文件的内容清除！因此若是在遍历文件的时候，你想同时修改文件就不能简单的以w只写模式打开了。

• 在遍历文件的时候想同时修改文件怎么办呢？注意这里说的是修改，好比修改某一行。

修改文件的某一部分

这里举例在遍历文件的时候想修改文件，好比以上面遍历目录文件的例子：

def list_file_by_walk(dir_path):
    for home, dirs, files in os.walk(dir_path):
        print(home)
        print(dirs)
        # files中是文件列表
        for file_name in files:
            # 这个file_name只是一个文件名
            print(file_name)
            # 若是咱们须要打开文件进行编辑读写操做，那就须要文件的路径
            # 用os.path.join方法就能够快速获得文件路径
            file_path = os.path.join(home, file_name)
            
            # 这个时候我想修改这个文件的内容，要怎么作呢？
复制代码

须要注意的是，读写模式须要管理文件指针，比较麻烦，写读模式,则会把原先的内容都清除，因此都不适合

一种思路是我先以只读模式打开，而后遍历文件的内容保存起来，好比：

with open(file, 'r') as f:
    for line in f.readlines():
        print line
        # 这里就能够把文件的内容一行行保存起来
        # 找到须要修改的行，修改之后保存
复制代码

保存之后再以只读模式打开文件，写入便可。

另外一种思路是我打开原文件的同时，再打开另外一个临时文件用于写入，好比这样：

with open(file, 'r') as f, open(cache_file_path, 'w') as w:
    for line in f.readlines():
        # 中间能够对f文件中的内容进行过滤或是修改
        w.write(line)

# 完了之后须要删除原文件，而后把临时文件的名字修改为原来文件的名字便可
复制代码

这里就顺道引出了，删除和重命名文件的方法

os.remove(file)
os.rename(cache_file_path, file)
复制代码

总结

以上就是Python文件和目录操做的经常使用知识点和方法总结，建议你们收藏起来，之后方便随时查看。

---

欢迎关注个人公众号查看更多精彩文章！
复制代码