python输入一个文件地址后删除指定后缀文件
时间: 2024-09-17 19:05:07 浏览: 42
当你想在Python中读取一个文件并删除它的特定后缀时,你可以使用`os`模块来操作文件路径,然后使用`shutil`模块中的`rmtree`或`remove`函数来删除文件。以下是一个简单的示例:
```python
import os
def delete_file_with_extension(input_path, extension):
# 获取文件名(不包括扩展名)
base_name = os.path.splitext(os.path.basename(input_path))[0]
# 去除原始路径,只留下文件名
file_to_delete = os.path.join(os.path.dirname(input_path), base_name)
# 检查是否为目录还是单个文件
if os.path.isdir(file_to_delete):
# 如果是目录,先删除目录中的所有文件,再删除目录本身
for filename in os.listdir(file_to_delete):
file_to_remove = os.path.join(file_to_delete, filename)
try:
os.remove(file_to_remove) # 删除文件
except FileNotFoundError:
pass # 忽略找不到的文件
shutil.rmtree(file_to_delete) # 删除空目录
else:
os.remove(file_to_delete) # 直接删除文件
# 使用方法
input_address = input("请输入文件地址: ")
extension_to_remove = input("请输入要删除的后缀(如'.txt'): ")
delete_file_with_extension(input_address, extension_to_remove)
```
在这个示例中,用户被提示输入文件地址和他们希望删除的后缀。然后,程序会找到包含指定后缀的文件(如果是目录,则先删除其中所有文件),并将之删除。
阅读全文