创建Python函数以删除文件或目录

删除文件或目录是在程序中非常常见的操作之一。在Python中，我们可以使用os模块来执行文件和目录的删除操作。下面是一个用于删除文件或目录的Python函数的示例。

import os

def delete_file_or_dir(path):
    """
    删除指定的文件或目录
    
    参数:
    path (str): 文件或目录的路径
    
    返回值:
    无
    """
    if os.path.isfile(path):  # 检查是否是文件
        try:
            os.remove(path)  # 删除文件
            print(f"已成功删除文件：{path}")
        except OSError as e:
            print(f"删除文件时出错：{e}")
    elif os.path.isdir(path):  # 检查是否是目录
        try:
            os.rmdir(path)  # 删除目录
            print(f"已成功删除目录：{path}")
        except OSError as e:
            print(f"删除目录时出错：{e}")
    else:
        print(f"无效的路径：{path}")

# 示例使用
delete_file_or_dir("path/to/file.txt")  # 删除文件
delete_file_or_dir("path/to/dir")  # 删除目录

在这个函数中，我们首先使用os.path.isfile(path)和os.path.isdir(path)分别检查指定路径是否是文件或目录。然后分别调用os.remove(path)和os.rmdir(path)来删除文件或目录。

注意，使用rmdir删除目录时，目录必须为空。如果目录非空，可以使用shutil.rmtree(path)来删除目录及其所有内容。

希望这个简单的示例函数能帮助你理解如何在Python中删除文件或目录。当在实际项目中使用时，请确保在删除文件或目录之前进行适当的权限和错误检查，以避免不必要的问题。