根据其内容重命名列

Question

我看到了一些类似的问题，但不是我要找的问题。

我有一张交付给我的 excel 表：

• 我需要的列在那里
• 列名在第二行
• 但列的顺序可能会有所不同（有时数字在前，或者城市和名称的顺序不同）
• 有 2 列同名。

这是如何复制：

data = [['name','name','number'], ['Nick','Brussels', 15], ['Tom','Paris', 14]]
df = pd.DataFrame(data)

我让我的第一行列标题是这样的：

df=df.rename(columns=df.iloc[0]).drop(df.index[0])

我可以肯定的是，“布鲁塞尔”的值始终在数据集中，所以我的问题是：我可以根据该列中的值更改列的名称。

所以在伪代码中这将是：

if the column contains the word 'Brussels' rename the column to 'city' 

到目前为止，我所有的尝试都更改了两个列名，因为当我选择包含布鲁塞尔的列时，它返回“名称”作为要更改的索引。 我想要iloc，退回...

我的目标是拥有这个：

   name      city number
1  Nick  Brussels     15
2   Tom     Paris     14

Answer 1

您可以使用布尔掩码来查找正确的列：

df.columns = [c if not m else 'city'
                  for c, m in zip(df.columns, df.eq('Brussels').any())]
print(df)

# Output
   name      city number
1  Nick  Brussels     15
2   Tom     Paris     14

Answer 2

这类似于 Corralien 在使用布尔掩码时的回答，但它不是列表理解，而是首先修改源行，然后像您一样将其分配为列。

df.iloc[0][df.eq('Brussels').any()] = 'city'
df = df.rename(columns=df.iloc[0]).drop(df.index[0])