博客

Python中os模块的12种用法总结

  • Post category:Python

Python中os模块的12种用法总结

Python中的OS模块提供了许多方法来处理与操作系统相关的任务，包括文件和目录操作、进程管理、环境变量、获取系统信息等等。在本篇文章中，我们将总结一下Python中os模块的12种主要用法及示例，以便更好地使用该模块。

1. 获取当前工作目录

通过os模块可以获得当前Python脚本文件所在的工作目录，使用方法如下：

import os
cwd = os.getcwd()
print(cwd)

输出结果为当前的工作目录。

2. 判断文件或目录是否存在

使用os模块还可以方便地判断文件夹或文件是否存在，如果存在，返回True，否则返回False。

示例：

import os
isExists = os.path.exists("./Test")    
print(isExists)

上述示例将返回是否存在文件夹Test，输出结果为True或者False。

3. 创建文件夹

如果需要在Python中创建新文件夹，可以使用以下语句：

import os
if not os.path.exists('./Test'):
    os.makedirs('./Test')

上述示例将新建一个名为Test的文件夹。

4. 获取目录下所有文件和文件夹

在Python中，可以使用os.listdir()方法来获取所有文件和文件夹的列表，将其存放在一个列表中，例如：

import os
fileList = os.listdir('./Test')
print(fileList)

上述例子获取名为Test的文件夹下的所有文件和文件夹列表。

5. 复制文件或者目录

把一个文件或者一整个目录复制到另一个地方。使用os.system()方法来实现：

import os
os.system('cp -r ./Test ./CopyOfTest')

上述示例将文件夹Test拷贝到名为CopyOfTest的新文件夹中。

6. 重命名文件或者目录

我们可以使用os.rename()方法实现重命名文件或者文件夹，例如：

import os
os.rename('old_filename.txt', 'new_filename.txt')

上述语句将old_filename.txt重命名为new_filename.txt。

7. 删除文件或者目录

删除文件或者整个目录操作在开发中经常使用。调用os.remove()方法删除单一文件，调用os.rmtree()方法可以删除整个目录：

import os
os.remove('./test.txt')
os.rmtree('./Test')

上述两例将删除一个名为test.txt的文件和文件夹Test。

8. 获取环境变量

Python中可以使用os.environ方法获取当前环境中的环境变量。

import os
env = os.environ
print(env)

上述代码将会打印当前环境变量。

9. 获取文件信息

通过使用os.stat()方法可以获取文件的信息，例如：

import os
file_info = os.stat('./test.txt')
print(file_info)

上述代码将会返回文件名、目录路径、大小、创建时间等相关信息。

10. 获得文件夹大小

获得目录的大小常常是一个优化操作过程，可以使用以下代码：

import os
path = './Test'
print(sum(os.path.getsize(os.path.join(path, f)) for f in os.listdir(path)))

上述例子假设在名为Test的文件夹中有若干文件，在获取每个文件大小后将其相加，最后返回文件夹的总大小。

11. 管理文件权限

Python中通过os.chmod()方法可以修改文件权限。示例如下：

import os
os.chmod("./test.txt", 0o777)

上述示例将文件test.txt的访问权限改为最大，即任何人均可访问。

12. 获取文件名和文件扩展名

使用os.path.splitext()方法可以方便地获取一个文件的名称和扩展名。

import os
file_path = "./test.txt"
(file_name, ext) = os.path.splitext(file_path)
print(file_name + "\n" + ext)

上述代码将会输出文件名和扩展名。

本文总结了Python中os模块的12种用法，并给出了每一种用法的详细示例。希望可以帮助大家更快掌握该模块的使用方法。

Tags: