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

Question

I want to get column name from the whole database (assume the database contains more than 100 rows with more than 50 column) based on specific value that contain in a specific column in pandas.

Here is my code:

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)

Tried a lot . . .

Example: when the user will input 2; it will show answer: A if the input is 4; feedback will be B and if 7 then reply will be C

tried iloc but I've seen row have to be noticed there.

Please Help Dear Guys . . . . . Thanks . . . :)

Answer 1

First of all you should treat the input as integer. So instead of raw_input , use input :

response = input("input")

After that you can use any :

df[df==YOUR_VALUE].any()

This will return a boolean Series with columns names and whether they contain the value you are looking for.

In your example:

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

for input 4 the output will be:

A    False
B     True
C    False
dtype: bool

Answer 2

Try this

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