在 Python 中,有时候需要删除文件或文件夹,可以使用 Python 的 os 模块或 shutil 模块来实现。下面我们分别来介绍这两种方式的具体实现。
1. 使用 os 模块
使用 os 模块删除文件可以使用 os.remove() 方法,删除空文件夹可以使用 os.rmdir() 方法,但是如果文件夹中还有其他文件或文件夹,则无法删除,需要先删除子元素,再删除空文件夹。可以使用 os.path.exists() 方法来判断文件或文件夹是否存在。
下面是一个删除文件的示例:
import os
if os.path.exists('example.txt'):
os.remove('example.txt')
print('文件删除成功')
else:
print('文件不存在')
下面是一个删除文件夹的示例:
import os
if os.path.exists('example'):
for dirpath, dirnames, filenames in os.walk('example'):
for filename in filenames:
os.remove(os.path.join(dirpath, filename))
for dirname in dirnames:
os.rmdir(os.path.join(dirpath, dirname))
os.rmdir('example')
print('文件夹删除成功')
else:
print('文件夹不存在')
2. 使用 shutil 模块
shutil 模块是 Python 自带的文件、文件夹操作模块,包括复制、移动、重命名和删除。比 os 模块更加高级和方便。使用 shutil.rmtree() 方法可以删除非空文件夹,该方法会遍历整个文件夹,删除所有文件和文件夹。可以使用 os.path.isdir(), os.path.isfile() 等方法来判断文件或文件夹是否存在。
下面是一个删除文件的示例:
import shutil
if os.path.isfile('example.txt'):
os.remove('example.txt')
print('文件删除成功')
else:
print('文件不存在')
下面是一个删除文件夹的示例:
import shutil
if os.path.exists('example'):
shutil.rmtree('example')
print('文件夹删除成功')
else:
print('文件夹不存在')
总结
以上就是使用 Python 删除文件或文件夹的两种方法,可以根据需要选择使用 os 模块或 shutil 模块。需要注意的是,在进行删除操作时,一定要确认要删除的文件或文件夹存在,否则会出现错误。