pandas获得某一值对应的列名
时间: 2023-08-15 21:09:42 浏览: 208
Python Pandas 获取列匹配特定值的行的索引问题
要获得某一值对应的列名,你可以使用pandas的`idxmax()`函数。这个函数可以返回每行或每列中最大值所在的索引位置。下面是一个示例代码:
```python
import pandas as pd
# 创建一个示例DataFrame
data = {'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 查找值为5的列名
col_name = df.columns[df.eq(5).any()]
print(col_name)
```
在这个例子中,我们创建了一个DataFrame,并查找值为5的列名。`df.eq(5).any()`会返回一个布尔型的Series,表示每列中是否存在值为5的元素。然后,我们使用`df.columns[]`来获取列名对应的索引位置,最后打印出结果。
输出结果将是:
```
Index(['B'], dtype='object')
```
这意味着值为5的元素在'B'列中。
阅读全文