Python 标准库 (os)：最强文件操作工具，你绝对不能错过！
Python 提供了一个非常强大的标准库，叫做 os 模块，它为我们提供了一系列强大的函数来与操作系统进行交互，特别是在文件和目录的操作方面。无论你是新手还是经验丰富的程序员，os 模块都能大大提升你的开发效率。在这篇文章中，我们将通过一些简单易懂的例子，带你深入了解如何用 os 来处理文件、目录以及与操作系统的其他交互。
1. os 模块概述
os 模块是 Python 提供的一个标准库模块，它提供了与操作系统进行交互的功能，包括文件和目录操作、进程管理、环境变量访问等。借助 os，你可以轻松实现文件路径的操作、创建目录、删除文件等操作，甚至可以通过它执行 shell 命令。
导入 os 模块
要使用 os 模块，首先需要将其导入：
import os
一旦导入，你就可以使用其中的各种函数了。
2. 常用 os 函数
2.1 获取当前工作目录
os.getcwd() 用于获取当前的工作目录。当前工作目录就是你运行 Python 程序时所在的文件夹。
import os
# 获取当前工作目录
current_directory = os.getcwd()
print("当前工作目录是:", current_directory)
2.2 改变工作目录
如果你需要切换到另一个目录，可以使用 os.chdir(path) 来改变当前工作目录。path 是你想要切换到的目录路径。
import os
# 切换到上一级目录
os.chdir('..')
print("当前工作目录是:", os.getcwd())
2.3 创建新目录
使用 os.mkdir(path) 可以创建一个新目录。如果目录已经存在，将会抛出 FileExistsError 错误。
import os
# 创建一个名为 'new_folder' 的目录
if not os.path.exists('new_folder'):
os.mkdir('new_folder')
print("目录 'new_folder' 已创建")
else:
print("目录 'new_folder' 已存在")
2.4 删除目录
可以使用 os.rmdir(path) 来删除一个空目录。如果目录不为空，将会抛出 OSError 错误。
import os
# 删除目录 'new_folder'
if os.path.exists('new_folder'):
os.rmdir('new_folder')
print("目录 'new_folder' 已删除")
else:
print("目录 'new_folder' 不存在")
2.5 列出目录中的文件和子目录
你可以使用 os.listdir(path) 来列出指定目录下的所有文件和子目录。这个函数会返回一个包含文件名和目录名的列表。
import os
# 列出当前工作目录下的所有文件和目录
files = os.listdir('.')
print("当前目录中的文件和目录有:", files)
2.6 判断路径类型
os.path 子模块提供了许多与路径相关的功能。例如，os.path.exists(path) 可以判断一个路径是否存在，os.path.isdir(path) 可以判断一个路径是否是目录，os.path.isfile(path) 用于判断路径是否是文件。
import os
# 判断路径类型
path = 'example.txt'
if os.path.exists(path):
if os.path.isdir(path):
print(f"{path} 是一个目录")
elif os.path.isfile(path):
print(f"{path} 是一个文件")
else:
print(f"{path} 路径不存在")
2.7 获取文件大小
可以通过 os.path.getsize(path) 获取文件的大小，单位为字节。
import os
# 获取文件 'example.txt' 的大小
file_path = 'example.txt'
if os.path.exists(file_path) and os.path.isfile(file_path):
file_size = os.path.getsize(file_path)
print(f"{file_path} 的文件大小是 {file_size} 字节")
else:
print(f"{file_path} 文件不存在")
2.8 执行系统命令
通过 os.system(command)，你可以在 Python 程序中执行操作系统命令。这对于一些需要和外部程序交互的任务非常有用。
import os
# 执行一个简单的系统命令
os.system('echo Hello, world!')
这个命令会在控制台输出
Hello, world!，在 Windows 和类 Unix 系统上都能运行。
3. os 模块的高级功能
3.1 环境变量
os 模块还提供了对环境变量的访问，可以使用 os.environ 来获取和修改环境变量。os.environ 返回一个字典对象，其中包含了所有环境变量。
import os
# 获取一个环境变量
path_env = os.environ.get('PATH')
print("系统的 PATH 环境变量是:", path_env)
你也可以使用 os.environ 来修改环境变量。例如：
import os
# 设置新的环境变量
os.environ['MY_VARIABLE'] = 'my_value'
print("新设置的环境变量 MY_VARIABLE 是:", os.environ['MY_VARIABLE'])
3.2 临时文件和目录
os 模块还提供了一些用于创建临时文件和目录的工具，如 os.mkdtemp() 和 os.tempnam()。这些函数可以帮助你在运行时创建临时的文件和目录，适用于测试和临时存储数据。
import os
# 创建临时目录
temp_dir = os.mkdtemp()
print("临时目录路径是:", temp_dir)
4. 总结
Python 的 os 模块是与操作系统交互的强大工具，涵盖了文件和目录操作、环境变量管理、执行系统命令等众多功能。通过本文的简单例子，我们了解了如何使用 os 模块来轻松处理文件和目录操作，并且介绍了 os 模块的一些高级功能。
无论你是进行文件管理、脚本编写，还是需要执行系统命令，os 模块都能为你提供最强大的支持。掌握它，能够让你的 Python 开发更加高效、灵活！
原文始发于微信公众号（小陈大看点）：Python 标准库 (os)：最强文件操作工具，你绝对不能错过！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/311308.html
