根据其内容重命名列

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])