Python清空目录

Python清空目录

在使用Python进行文件处理时，有时候我们可能需要清空一个目录中的所有文件和子目录。本文将详细介绍如何使用Python清空指定目录的方法。

方法一：使用os模块和shutil模块

我们可以使用os模块和shutil模块来清空目录。首先，我们需要导入这两个模块：

import os
import shutil

接下来，我们定义一个函数来清空目录。该函数会先删除目录下的所有文件，然后递归删除目录下的所有子目录。

def clear_directory(path):
    for file_name in os.listdir(path):
        file_path = os.path.join(path, file_name)

        if os.path.isdir(file_path):
            shutil.rmtree(file_path)
        else:
            os.remove(file_path)

最后，我们可以调用这个函数来清空指定目录。比如，我们可以清空当前目录下的test目录：

path = 'test'
clear_directory(path)
print(f'{path} directory has been cleared.')

运行结果将输出：

test directory has been cleared.

方法二：使用os模块和os.path模块

除了使用shutil模块外，我们还可以使用os模块和os.path模块来清空目录。同样地，首先我们需要导入这两个模块：

import os
import os.path

然后，我们定义一个函数来清空目录。与方法一不同的是，该函数会使用os.path.isdir()检查是否为目录，并使用os.path.isfile()检查是否为文件。

def clear_directory(path):
    for file_name in os.listdir(path):
        file_path = os.path.join(path, file_name)

        if os.path.isdir(file_path):
            for sub_file in os.listdir(file_path):
                sub_file_path = os.path.join(file_path, sub_file)
                if os.path.isdir(sub_file_path):
                    shutil.rmtree(sub_file_path)
                else:
                    os.remove(sub_file_path)
            shutil.rmtree(file_path)
        else:
            os.remove(file_path)

最后，我们同样可以调用这个函数来清空指定目录，例如清空当前目录下的test目录：

path = 'test'
clear_directory(path)
print(f'{path} directory has been cleared.')

运行结果将输出：

test directory has been cleared.

总结

本文介绍了使用Python清空目录的两种方法：一种是使用os模块和shutil模块，另一种是使用os模块和os.path模块。根据实际情况，我们可以选择合适的方法来清空目录。