Python 如何按单元格值获取dataframe列索引

我有一个如下所示的数据帧：

如何根据单元格值exist

.xml

获取列索引，该值在我的数据框中为4

这是我的密码：

df = fileserver
for index in df:
    df1 = df[index].str.contains(".xml")
    print(df1)

---

对于结果

True

，我不知道如何找到索引，请帮助并感谢大家您可以使用以下代码跟踪哪个列确实包含模式

.xml

。对于下面的示例，感兴趣的列具有索引2

首先，为您的示例创建一个示例数据帧

import pandas as pd

>>> df = pd.DataFrame([[1,2,"3.xml"], [4,5,"6.xml"]])  
>>> df
   0  1      2
0  1  2  3.xml
1  4  5  6.xml

现在，在数据帧的所有单元格中检查您的模式

>>> df_filtered = df.applymap(lambda cell: str(cell).endswith(".xml")) 
>>> df_filtered
       0      1     2
0  False  False  True
1  False  False  True

最后，如果至少找到一个具有所需模式的单元格，则保留列

>>> [column for column, count in df_filtered.sum().to_dict().items() 
     if count > 0]
[2]

---

谢谢你救了我的命（）