Python 删除目录下所有文件|极客笔记

Python 删除目录下所有文件

在日常的工作中，有时候我们需要删除某个目录下所有的文件。虽然可以手动一个一个去删除，但是这样费时费力。在Python中，可以通过操作系统模块os来实现这个任务。本文将详细介绍如何使用Python来删除目录下所有的文件。

1. 确保目录存在

在删除目录下所有文件之前，首先要确保目录存在。可以通过os.path.exists来判断目录是否存在，如果不存在则创建目录。

import os

directory = 'test_dir'

if not os.path.exists(directory):
    os.makedirs(directory)

2. 获取目录下所有文件

接下来，我们需要获取目录下所有的文件。可以使用os.listdir来获取目录下的所有文件，然后遍历所有文件，使用os.path.join来获取文件的绝对路径。

import os

directory = 'test_dir'

files = os.listdir(directory)

for file in files:
    file_path = os.path.join(directory, file)
    print(file_path)

运行以上代码，可以列出test_dir目录下的所有文件的绝对路径。

3. 删除目录下所有文件

有了目录下所有文件的绝对路径后，就可以使用os.remove来删除这些文件了。可以通过遍历所有文件路径，对每一个文件进行删除操作。

import os

directory = 'test_dir'

files = os.listdir(directory)

for file in files:
    file_path = os.path.join(directory, file)
    os.remove(file_path)

运行以上代码，可以删除test_dir目录下所有的文件。需要注意的是，如果目录下还有子目录，需要先删除子目录下的文件，再删除子目录本身。

4. 完整代码

综合以上步骤，可以写一个完整的Python脚本来删除目录下所有文件。

import os

def remove_files(directory):
    if not os.path.exists(directory):
        os.makedirs(directory)

    files = os.listdir(directory)

    for file in files:
        file_path = os.path.join(directory, file)
        os.remove(file_path)

remove_files('test_dir')

以上代码将删除test_dir目录下所有的文件。