查找数据框中某行最大值的列名

Question

I am using a pandas data frame in Python and have 121 columns and 5 rows. I want to find the max value in each row and print the name of the column where it occurs. I really have no idea how I would even begin to do this but I would appreciate the help.

Edit: the column names are strings

Edit: Some example scenario is:

and for this example I would want something like:

id: 000621fb3cbb32d8935728e48679680e
breed: african_hunting_dog
confidence: 0.011319

etc for all 3 rows

Answer 1

You can using idxmax and max with axis=1

df.idxmax(axis=1)
df.max(axis=1)

Answer 2

You may try this for beginner:

import pandas as pd

df = pd.DataFrame({'Col1':['A','B','C','D','E'], 'Col2': [4,2,6,3,2], 'Col3':[3,6,4,3,1]})
for x in range(len(df)):
    dt = df.iloc[x]
    if dt[1] > dt[2]:
        print ('Col2:{}'.format(dt[1]))
    elif dt[1] < dt[2]:
        print ('Col3:{}'.format(dt[2]))
    else:
        print ('Both cols are same')

Answer 3

    for col in df.columns:
       if df[col].values[-1] == df.iloc[-1].max():
          print(col)

or alternatively

    [x if df[x].values[-1] == df.iloc[-1].max() for x in df.columns]