Python学习之os模块及用法

在Python中，os模块是常用的操作系统接口之一，它允许我们执行各种操作系统相关的任务。本文将介绍os模块的使用方法及一些常见的用法。

os模块的导入

在使用os模块之前，需要导入它。通过以下语句导入os模块：

import os

os模块常见操作

获取当前工作目录

我们可以使用以下方法来获取当前工作目录：

os.getcwd()

例如，我们可以这样使用该方法：

import os

# 获取当前工作目录
current_dir = os.getcwd()

# 输出当前工作目录
print("当前工作目录为: ", current_dir)

输出结果如下：

当前工作目录为: /Users/john/Documents

列出目录内容

os模块的listdir()方法可以列出目录中的所有文件和子目录：

os.listdir(path)

其中，path参数可选，默认为当前工作目录。
例如，我们可以这样使用该方法：

import os

# 列出当前目录下的所有文件和目录
content = os.listdir()

# 输出目录内容
print("当前目录内容为：")
for c in content:
    print(c)

输出结果如下：

当前目录内容为：
file1.txt
file2.txt
subdir1
subdir2

创建目录

我们可以使用os模块的mkdir()方法来创建新的目录：

os.mkdir(path)

例如，我们可以这样使用该方法：

import os

# 创建一个新目录
os.mkdir("newdir")

# 输出提示信息
print("目录创建成功")

删除目录

我们可以使用os模块的rmdir()方法来删除目录：

os.rmdir(path)

其中，path为要删除的目录路径。
例如，我们可以这样使用该方法：

import os

# 删除一个目录
os.rmdir("newdir")

# 输出提示信息
print("目录删除成功")

示例说明

示例1：列出指定目录中所有文件

import os

# 指定目录路径
dir_path = "/Users/john/Documents"

# 列出目录下的所有文件
content = os.listdir(dir_path)

# 输出文件列表
print("目录 {} 下的所有文件：".format(dir_path))
for c in content:
    # 判断是否为文件
    if os.path.isfile(os.path.join(dir_path, c)):
        print(c)

输出结果如下：

目录 /Users/john/Documents 下的所有文件：
file1.txt
file2.txt

示例2：列出子目录中所有文件

import os

# 指定目录路径
dir_path = "/Users/john/Documents"

# 列出目录下的所有子目录
subdirs = [d for d in os.listdir(dir_path) if os.path.isdir(os.path.join(dir_path, d))]

# 列出所有子目录中的所有文件
for subdir in subdirs:
    content = os.listdir(os.path.join(dir_path, subdir))
    print("子目录 {} 下的所有文件：".format(subdir))
    for c in content:
        # 判断是否为文件
        if os.path.isfile(os.path.join(dir_path, subdir, c)):
            print(c)

输出结果如下：

子目录 subdir1 下的所有文件：
file3.txt
file4.txt
子目录 subdir2 下的所有文件：
file5.txt
file6.txt