在Python中，os模块提供了与操作系统进行交互的功能，允许我们执行各种操作系统任务，如文件和目录操作、环境变量访问、进程管理等。os模块是标准库的一部分，无需额外安装。本文将详细介绍os模块的功能，并提供相应的示例代码，帮助全面掌握这一强大工具。
os 模块概述
os模块提供了一组函数和变量，使用Python代码执行操作系统级别的操作。
通过os模块，可以完成以下任务：
- 文件和目录操作（创建、删除、移动、重命名等）
- 环境变量访问
- 执行系统命令
- 进程管理
- 获取系统信息
文件和目录操作
获取当前工作目录
使用os.getcwd()函数可以获取当前工作目录。
import os
current_directory = os.getcwd()
print(f"当前工作目录: {current_directory}")
改变工作目录
使用os.chdir()函数可以改变当前工作目录。
import os
new_directory = "/path/to/new/directory"
os.chdir(new_directory)
print(f"已更改工作目录: {os.getcwd()}")
创建目录
使用os.mkdir()函数可以创建一个新目录。
import os
new_directory = "new_folder"
os.mkdir(new_directory)
print(f"已创建目录: {new_directory}")
使用os.makedirs()可以创建多级目录。
import os
new_directories = "new_folder/sub_folder"
os.makedirs(new_directories)
print(f"已创建多级目录: {new_directories}")
删除目录
使用os.rmdir()可以删除一个空目录。
import os
directory_to_remove = "new_folder/sub_folder"
os.rmdir(directory_to_remove)
print(f"已删除目录: {directory_to_remove}")
使用os.removedirs()可以删除多级空目录。
import os
directories_to_remove = "new_folder/sub_folder"
os.removedirs(directories_to_remove)
print(f"已删除多级目录: {directories_to_remove}")
列出目录内容
使用os.listdir()可以列出指定目录中的内容。
import os
directory = "."
contents = os.listdir(directory)
print(f"{directory} 目录内容:")
for item in contents:
print(item)
检查路径
使用os.path模块可以检查路径是否存在以及路径的类型（文件或目录）。
import os
path = "example.txt"
if os.path.exists(path):
if os.path.isfile(path):
print(f"{path} 是一个文件")
elif os.path.isdir(path):
print(f"{path} 是一个目录")
else:
print(f"{path} 不存在")
重命名和移动文件
使用os.rename()可以重命名或移动文件和目录。
import os
source = "example.txt"
destination = "renamed_example.txt"
os.rename(source, destination)
print(f"已将 {source} 重命名为 {destination}")
删除文件
使用os.remove()可以删除文件。
import os
file_to_remove = "renamed_example.txt"
os.remove(file_to_remove)
print(f"已删除文件: {file_to_remove}")
环境变量访问
获取环境变量
使用os.environ可以访问系统环境变量。
import os
path = os.environ.get('PATH')
print(f"系统 PATH 环境变量: {path}")
设置环境变量
使用os.environ可以设置系统环境变量。
import os
os.environ['MY_VAR'] = '12345'
print(f"已设置环境变量 MY_VAR: {os.environ.get('MY_VAR')}")
执行系统命令
使用os.system()可以执行系统命令。
import os
os.system('echo Hello, World!')
进程管理
获取当前进程ID
使用os.getpid()可以获取当前进程ID。
import os
pid = os.getpid()
print(f"当前进程ID: {pid}")
获取父进程ID
使用os.getppid()可以获取父进程ID。
import os
ppid = os.getppid()
print(f"父进程ID: {ppid}")
创建子进程
使用os.fork()可以创建子进程（仅适用于类Unix系统）。
import os
pid = os.fork()
if pid == 0:
print("这是子进程")
else:
print(f"这是父进程，子进程ID: {pid}")
获取系统信息
获取操作系统名称
使用os.name可以获取操作系统名称。
import os
os_name = os.name
print(f"操作系统名称: {os_name}")
获取平台名称
使用os.uname()可以获取系统的详细信息（仅适用于类Unix系统）。
import os
system_info = os.uname()
print(f"系统信息: {system_info}")
综合示例
以下是一个综合示例，展示如何使用os模块进行一系列操作。
import os
# 获取并打印当前工作目录
current_directory = os.getcwd()
print(f"当前工作目录: {current_directory}")
# 创建新目录
new_directory = "test_folder"
os.mkdir(new_directory)
print(f"已创建目录: {new_directory}")
# 改变工作目录
os.chdir(new_directory)
print(f"已更改工作目录: {os.getcwd()}")
# 创建新文件并写入内容
with open("test_file.txt", "w") as file:
file.write("Hello, World!")
print("已创建文件: test_file.txt 并写入内容")
# 列出当前目录内容
contents = os.listdir(".")
print(f"当前目录内容: {contents}")
# 重命名文件
os.rename("test_file.txt", "renamed_test_file.txt")
print("已将文件重命名为: renamed_test_file.txt")
# 删除文件
os.remove("renamed_test_file.txt")
print("已删除文件: renamed_test_file.txt")
# 返回上一级目录并删除目录
os.chdir("..")
os.rmdir(new_directory)
print(f"已删除目录: {new_directory}")
总结
本文详细介绍了Python的os模块的各种功能，包括文件和目录操作、环境变量访问、执行系统命令、进程管理以及获取系统信息。通过具体的示例代码，展示了如何使用这些功能来进行操作系统级别的任务。无论是获取当前工作目录、创建和删除目录、重命名和移动文件，还是访问环境变量、执行系统命令、管理进程，os模块都提供了简洁而强大的接口。掌握os模块的使用，可以大大提高在Python编程中的效率和灵活性，在处理系统相关任务时游刃有余。