python - 获取 pandas 中包含特定值的列名

Question

我想根据 pandas 中特定列中包含的特定值从整个数据库中获取列名称(假设数据库包含超过 100 行和超过 50 列)。

这是我的代码:

import pandas as pd
df = pd.DataFrame({'A':[1,2,3], 'B':[4,5,6], 'C':[7,8,9]})
pos = 2
response = raw_input("input")
placeholder = (df == response).idxmax(axis=1)[0]
print df
print (placeholder)

尝试了很多。 。 .

示例:当用户输入2时；它将显示答案:A如果输入是4；反馈将是B如果是 7 那么回复将是 C

尝试过 iloc，但我发现必须注意那里的行。

请帮助亲爱的大家。 。 。 。 。谢谢 。 。 。 :)

Answer 1

试试这个

for i in df.columns:
    newDf = df.loc[lambda df: df[i] == response]
    if(not newDf.empty):
        print(i)