获取在熊猫中包含特定值的列名

Question

我想从整个数据库中获取列名（假设数据库包含超过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； 它将显示答案：如果输入为4，则为A；否则为0。 反馈将为B，如果为7，则回复为C

尝试过iloc，但我已经看到必须在那儿注意行。

请各位朋友帮助。 。 。 。 。 谢谢 。 。 。 :)

Answer 1

首先，您应该将输入视为整数。 因此，请使用input代替raw_input ：

response = input("input")

之后，您可以使用any ：

df[df==YOUR_VALUE].any()

这将返回带有列名称以及它们是否包含您要查找的值的布尔Series 。

在您的示例中：

df = pd.DataFrame({'A':[1,2,3], 'B':[4,5,6], 'C':[7,8,9]})
response = input("input")
placeholder = df[df==response].any()

对于输入4，输出将是：

A    False
B     True
C    False
dtype: bool

Answer 2

尝试这个

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