当前位置:知识文库 ❯ 图文

Python os模块

一、os模块概述

os模块是Python的标准库之一，提供了与操作系统进行交互的功能。它允许我们执行各种文件和目录操作，包括创建、删除、重命名、遍历等。该模块在不同操作系统（Windows、Linux、macOS）上都能正常工作，会自动适配不同平台的路径分隔符和系统特性。

使用os模块前，需要先导入它：

代码示例

import os

os模块中包含了一个重要的子模块os.path，专门用于处理路径相关的操作，如拼接路径、获取文件名、判断路径是否存在等。

二、常用语法与函数

os模块提供了丰富的函数用于文件和目录操作，以下是常用的核心函数：

目录操作函数

os.mkdir(path)：创建单个目录
os.makedirs(path)：递归创建多级目录
os.rmdir(path)：删除空目录
os.removedirs(path)：递归删除空目录
os.listdir(path)：列出目录中的所有文件和子目录

路径操作函数

os.path.join(path, *paths)：智能拼接路径
os.path.exists(path)：判断路径是否存在
os.path.isfile(path)：判断是否为文件
os.path.isdir(path)：判断是否为目录
os.path.basename(path)：获取路径中的文件名部分
os.path.dirname(path)：获取路径中的目录部分
os.path.abspath(path)：获取绝对路径

文件操作函数

os.remove(path)：删除文件
os.rename(src, dst)：重命名文件或目录
os.stat(path)：获取文件状态信息

遍历函数

os.walk(top)：递归遍历目录树，返回(root, dirs, files)三元组

三、基本用法详解

1. 创建和删除目录

使用os.mkdir()可以创建单个目录，但如果父目录不存在则会报错。此时需要使用os.makedirs()来递归创建多级目录。

代码示例

import os

# 创建单个目录
os.mkdir('test_dir')

# 创建多级目录（如果父目录不存在）
os.makedirs('parent/child/grandchild', exist_ok=True)

# 删除空目录
os.rmdir('test_dir')

# 递归删除空目录
os.removedirs('parent/child/grandchild')

2. 路径拼接与操作

使用os.path.join()可以智能拼接路径，它会根据操作系统自动使用正确的路径分隔符（Windows使用反斜杠，Linux/macOS使用正斜杠）。

代码示例

import os

# 智能拼接路径
path = os.path.join('home', 'user', 'documents', 'file.txt')
print(path)  # Windows: home\user\documents\file.txt

# 获取绝对路径
abs_path = os.path.abspath('file.txt')
print(abs_path)

# 获取文件名和目录名
print(os.path.basename('/home/user/file.txt'))  # file.txt
print(os.path.dirname('/home/user/file.txt'))   # /home/user

# 分割扩展名
print(os.path.splitext('script.py'))  # ('script', '.py')

3. 判断路径类型

在进行文件操作之前，通常需要判断路径是否存在以及路径的类型，避免操作不存在的路径引发异常。

代码示例

import os

path = 'test.txt'

# 判断路径是否存在
if os.path.exists(path):
    print('路径存在')

# 判断是否为文件
if os.path.isfile(path):
    print('这是一个文件')

# 判断是否为目录
if os.path.isdir(path):
    print('这是一个目录')

# 获取文件大小（字节）
size = os.path.getsize(path)
print(f'文件大小: {size} 字节')

4. 遍历目录

os.walk()是遍历目录树的强大工具，它会递归地遍历指定目录下的所有文件和子目录。

代码示例

import os

# 遍历目录树
for root, dirs, files in os.walk('my_project'):
    print(f'当前目录: {root}')
    print(f'子目录: {dirs}')
    print(f'文件: {files}')
    print('-' * 40)

四、完整代码示例

示例1：批量重命名文件

代码示例

import os

def batch_rename(directory, old_ext, new_ext):
    """批量修改文件扩展名"""
    for filename in os.listdir(directory):
        if filename.endswith(old_ext):
            # 构建旧路径和新路径
            old_path = os.path.join(directory, filename)
            new_filename = filename[:-len(old_ext)] + new_ext
            new_path = os.path.join(directory, new_filename)
            
            # 重命名文件
            os.rename(old_path, new_path)
            print(f'已重命名: {filename} -> {new_filename}')

# 使用示例：将所有.txt文件改为.md
# batch_rename('./documents', '.txt', '.md')

示例2：查找特定类型的文件

代码示例

import os

def find_files(directory, extension):
    """在目录中递归查找指定扩展名的文件"""
    found_files = []
    
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                full_path = os.path.join(root, file)
                found_files.append(full_path)
    
    return found_files

# 查找所有Python文件
# py_files = find_files('./my_project', '.py')
# for f in py_files:
#     print(f)

示例3：清理空目录

代码示例

import os

def remove_empty_dirs(directory):
    """递归删除目录树中的所有空目录"""
    for root, dirs, files in os.walk(directory, topdown=False):
        for dir_name in dirs:
            dir_path = os.path.join(root, dir_name)
            try:
                # 尝试删除目录，如果为空则成功，否则跳过
                os.rmdir(dir_path)
                print(f'已删除空目录: {dir_path}')
            except OSError:
                # 目录不为空，跳过
                pass

# remove_empty_dirs('./my_project')

示例4：获取文件详细信息

代码示例

import os
import time

def get_file_info(filepath):
    """获取文件的详细信息"""
    if not os.path.exists(filepath):
        print('文件不存在')
        return
    
    stat_info = os.stat(filepath)
    
    info = {
        '文件名': os.path.basename(filepath),
        '文件大小': f'{stat_info.st_size} 字节',
        '创建时间': time.ctime(stat_info.st_ctime),
        '修改时间': time.ctime(stat_info.st_mtime),
        '访问时间': time.ctime(stat_info.st_atime),
        '是否为文件': os.path.isfile(filepath),
        '是否为目录': os.path.isdir(filepath),
    }
    
    for key, value in info.items():
        print(f'{key}: {value}')

# get_file_info('example.txt')

示例5：安全的目录创建

代码示例

import os

def safe_create_dir(directory):
    """安全地创建目录，避免重复创建和权限错误"""
    try:
        if not os.path.exists(directory):
            os.makedirs(directory, exist_ok=True)
            print(f'目录创建成功: {directory}')
        else:
            print(f'目录已存在: {directory}')
    except PermissionError:
        print(f'权限不足，无法创建目录: {directory}')
    except OSError as e:
        print(f'创建目录时出错: {e}')

# safe_create_dir('./new_project/data')

五、注意事项与最佳实践

注意1：使用os.mkdir()创建目录时，如果父目录不存在会抛出FileNotFoundError异常。建议使用os.makedirs(path, exist_ok=True)来安全创建多级目录。

注意2：os.remove()只能删除文件，不能删除目录。删除目录需要使用os.rmdir()（仅限空目录）或shutil.rmtree()（删除非空目录）。

注意3：路径操作应始终使用os.path.join()而不是字符串拼接，以确保跨平台兼容性。手动拼接路径（如使用+）在不同操作系统上可能导致错误。

注意4：删除操作是不可逆的，执行删除前应仔细检查路径。建议先使用os.path.exists()验证路径，并使用try-except块捕获可能的异常。

小贴士

在Python 3.4+中，推荐使用pathlib模块替代os.path。pathlib提供了面向对象的API，使路径操作更加直观和简洁。但在处理需要跨版本兼容的代码时，os模块仍然是更好的选择。

六、os与pathlib对比

虽然os模块功能强大，但Python 3.4引入的pathlib模块提供了更现代的面向对象的文件路径操作方式。下面是两者的主要对比：

功能	os模块	pathlib模块
创建目录	os.makedirs(path)	Path(path).mkdir(parents=True)
路径拼接	os.path.join(a, b)	Path(a) / b
判断存在	os.path.exists(path)	Path(path).exists()
获取文件名	os.path.basename(path)	Path(path).name
获取扩展名	os.path.splitext(path)[1]	Path(path).suffix
遍历目录	os.walk(path)	Path(path).rglob('*')
API风格	函数式	面向对象

七、小结

os模块提供了与操作系统交互的接口，是Python文件操作的基础
os.path子模块专门处理路径相关操作，包括拼接、分割、判断等
os.walk()是递归遍历目录树的强大工具，返回(root, dirs, files)三元组
路径拼接应始终使用os.path.join()，避免手动拼接导致跨平台问题
删除操作是不可逆的，执行前应仔细验证路径并使用异常处理
Python 3.4+推荐使用pathlib模块进行路径操作，但os模块仍广泛使用

八、练习题

练习1

编写一个函数count_files_by_ext(directory)，统计指定目录下每种文件类型的数量。要求使用os.walk()遍历目录，返回一个字典，键为扩展名，值为文件数量。

练习2

编写一个函数organize_files(directory)，将指定目录中的所有文件按照扩展名分类，自动创建对应扩展名的子目录，并将文件移动到相应的子目录中。

常见问题

os.mkdir()和os.makedirs()有什么区别？

os.mkdir()只能创建单级目录，如果父目录不存在会抛出FileNotFoundError异常。而os.makedirs()可以递归创建多级目录，即使中间的父目录都不存在也能成功创建。建议使用os.makedirs(path, exist_ok=True)来安全创建目录。

如何删除一个非空目录？

os.rmdir()只能删除空目录。要删除非空目录，需要使用shutil模块的shutil.rmtree()函数。例如：import shutil; shutil.rmtree('path/to/directory')。这个操作是不可逆的，使用前请确保路径正确。

os.walk()返回的三元组分别是什么？

os.walk()返回一个生成器，每次迭代返回一个三元组：(dirpath, dirnames, filenames)。dirpath是当前遍历的目录路径（字符串），dirnames是该目录下的子目录列表，filenames是该目录下的文件列表。通过设置topdown=False参数可以从底层目录开始遍历。

为什么推荐使用os.path.join()而不是字符串拼接？

不同操作系统使用不同的路径分隔符：Windows使用反斜杠(\)，Linux和macOS使用正斜杠(/)。os.path.join()会自动根据当前操作系统选择正确的分隔符，确保代码的跨平台兼容性。手动拼接路径容易在不同系统上出现错误。

标签： os模块文件操作目录操作 os.path 路径操作 os.walk Python教程

本文涉及AI创作

内容由AI创作，请仔细甄别

list快速访问

poll相关推荐

Django静态文件管理教程：STATIC配置、collectstatic与Nginx部署

Django Migration迁移详解 - 数据库版本控制完整指南

Django ORM查询详解 - 数据库操作与优化完整指南

Django中间件详解 - 请求响应钩子系统完整指南

快速访问

相关推荐