python os操作文件

使用python来进行文件的操作。

1.路径、文件的检查

返回True代表存在。
检查给定的路径是否存在：

 os.path.exists(data_dir)

检查路径是否为文件：

os.path.isfile(data_dir)

检查路径是否为目录：

os.path.isdir(data_dir)

路径拼接：

os.path.join(root, sub_dir)
os.listdir(dir) #获取该路径下文件的列表

判断图片输入类型：

import imghdr
imghdr.what(img.path)

2.目录遍历

os.walk() 方法用于通过在目录树中游走输出在目录中的文件名，向上或者向下。

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

• top – 所要遍历的目录的地址,
• topdown --可选，默认为 True，则优先遍历 top 目录，否则自底向上进行遍历。
• onerror – 可选，需要一个 callable 对象，当 walk 需要异常时会调用。
• followlinks – 可选，如果为 True，则会遍历目录下的快捷方式(linux 下是软连接 symbolic link )实际所指的目录(默认关闭)，如果为 False，则优先遍历 top 的子目录。

返回的是一个三元组(root,dirs,files)。

• root 所指的是当前正在遍历的这个文件夹的本身的地址
• dirs 是一个 list ，内容是该文件夹中所有的目录的名字(不包括子目录)
• files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)

for root, dirs, files in os.walk(file):

    # root 表示当前正在访问的文件夹路径
    # dirs 表示该文件夹下的子目录名list
    # files 表示该文件夹下的文件list

    # 遍历文件
    for f in files:
       print(os.path.join(root, f))

    # 遍历所有的文件夹
    for d in dirs:
       print(os.path.join(root, d))

如何遍历，遍历的方式参考这篇文章：
https://zhuanlan.zhihu.com/p/149824829