如何基于DataFrame列名条件执行删行与列重命名操作？

云上老周

2026-8-10

问题解决：DataFrame列名匹配时的条件处理报错

需求

当DataFrame的列头为Unnamed: 0、VALUE、VALUE.1、VALUE.2时，删除前两行并重命名列头。尝试代码时因列名条件判断语法错误报错，需修正。

示例数据

Unnamed: 0       VALUE     VALUE.1        VALUE.2
Name            Hobbies     Dislikes    Favorite Color
Ben             NaN         NaN              NaN
Alex            NaN         Running          Red
Mike            NaN         Cartoons         Blue
Mark            NaN         Pizza            Yellow

错误代码

if df.columns = {"Unnamed: 0", "VALUE", "VALUE.1", "VALUE.2"}:
     df = df.drop([0,1])
     df = df.rename(columns={"Unnamed: 0": "Name", "VALUE": "Hobbies", "VALUE.1": "Dislikes", "VALUE.2": "Favorite Color"})

错误原因

条件判断误用赋值符号=，应使用比较符号==
df.columns是Pandas的Index对象，直接与集合比较无法得到正确匹配结果

正确代码

情况1：要求列名严格匹配顺序

if list(df.columns) == ["Unnamed: 0", "VALUE", "VALUE.1", "VALUE.2"]:
    # 删除前两行
    df = df.drop([0, 1])
    # 重命名列
    df = df.rename(columns={
        "Unnamed: 0": "Name", 
        "VALUE": "Hobbies", 
        "VALUE.1": "Dislikes", 
        "VALUE.2": "Favorite Color"
    })
    # 重置索引（可选，避免删除行后索引断层）
    df = df.reset_index(drop=True)

情况2：仅要求列名包含指定元素，不考虑顺序

if set(df.columns) == {"Unnamed: 0", "VALUE", "VALUE.1", "VALUE.2"}:
    df = df.drop([0, 1])
    df = df.rename(columns={
        "Unnamed: 0": "Name", 
        "VALUE": "Hobbies", 
        "VALUE.1": "Dislikes", 
        "VALUE.2": "Favorite Color"
    })
    df = df.reset_index(drop=True)