如何在Python中删除DataFrame中用户指定的多列？

要实现删除DataFrame中用户指定的多列，先指出你提供的代码存在几个关键问题：

• 循环时用range(len(s_col))取索引，传给df.drop的是数字索引而非用户输入的列名，会导致删除错误的列
• 异常捕获太宽泛，无法精准处理列不存在的情况
• 可以直接批量传入列名列表，无需逐个循环删除

以下是修正后的实现方案：

# 删除指定列
def drop_columns(df):
    # 获取用户输入的列名，按空格分隔
    s_col = input("请输入要删除的列名（用空格分隔）：").split()
    if not s_col:
        print("未输入任何列名，无需删除")
        print(df.info())
        return
    
    try:
        # 直接传入列名列表批量删除，inplace=True直接修改原DataFrame
        df.drop(s_col, axis=1, inplace=True)
        print(f"列 {', '.join(s_col)} 已成功删除")
    except KeyError as e:
        # 捕获列不存在的错误，告知用户具体哪些列不存在
        missing_cols = [col for col in s_col if col not in df.columns]
        print(f"列 {', '.join(missing_cols)} 不存在，未删除任何列")
    print(df.info())

关键说明： #

• 直接将用户输入的列名列表传给df.drop，axis=1指定操作列方向，inplace=True直接修改原DataFrame（如果不想修改原DataFrame，可以去掉inplace=True，接收返回的新DataFrame）
• 增加空输入判断，避免无效操作
• 精准捕获KeyError，并找出具体不存在的列，给用户明确反馈
• 用f-string替代format，语法更简洁直观

内容的提问来源于stack exchange，提问作者Gagan Ram