Python脚本中常用的OS模块函数汇总（持续更新）

---

常用函数

• 获取当前路径 os.getcwd()
• 判断文件或目录是否存在 os.path.exists
• 判断路径是文件形式 os.path.isfile
• 创建目录 os.mkdir
• 复制文件 shutil.copy
• 拼接路径 os.path.join 自动选择 \ 或 /
• 删除文件 os.remove
• 分离文件名和扩展名 os.path.splitext

获取文件夹下所有文件名

folder = "/home/llb/datasets/ImageNet2012/images/"
for root, subdirs, files in os.walk(folder, topdown=True):
    file_list = [ os.path.join(root, file) for file in files]

提取指定格式的所有文件

def find_files(folder: str, file_type: str):
    file_list = []
    for root, subdirs, files in os.walk(folder, topdown=True):
        """
        输出项中 root 是每个 file 的目录
        subdirs 是该目录下所有一级子目录的名称列表
        files 是该目录下所有文件的名称列表
        因此，文件路径为 os.path.join(root, file)
        """
        for file in files:
            file_path = os.path.join(root, file).replace('\\', '/')
            if file_type == file_path.split('.')[-1]:
                file_list.append(file_path)
    print(f"Total of {len(file_list)} {file_type} files have been found.")
    return file_list

合并文件，类似于cat命令

file = "train.txt"
if os.path.exists(file):
    os.remove(file)
with open(file, 'a') as w:   # 追加模式
    with open("2007_train.txt", 'r') as r:
        txt = r.read()
    w.write(txt)
    with open("2007_val.txt", 'r') as r:
        txt = r.read()
    w.write(txt)

win系统下复制文件 shutil.copy命令

使用高级文件接口：shutil

src = "./1.txt"
dst = "E:"
shutil.copy(src, dst)   # 会发生覆盖

不建议使用 os.system 命令：

"""
- 路径分割必须是反斜杠！！！
- 相对路径或绝对路径都可以，跨盘复制也可以
- 文件存在则覆盖
"""
src = ".\\1.txt"
dst = "E:"
sh = "copy " + src + " " + dst
os.system(sh)