/Python/Pandas find column with max value for each row

Pandas find column with max value for each row

pandas.set_option('display.max_rows', None)

# displays max number of columns and rows


import pandas as pd
pd.options.display.max_rows = 999
pd.options.display.max_columns = 999

>>> df.idxmax(axis=1)
0    Communications
1          Business
2    Communications
3    Communications
4          Business
dtype: object

Source: stackoverflow.com

# credit to Stack Overflow in the source link

df.head()
Communications and Search   Business    General Lifestyle
0   0.745763    0.050847    0.118644    0.084746
0   0.333333    0.000000    0.583333    0.083333
0   0.617021    0.042553    0.297872    0.042553
0   0.435897    0.000000    0.410256    0.153846
0   0.358974    0.076923    0.410256    0.153846

df.idxmax(axis=1)
0    Communications
1          Business
2    Communications
3    Communications
4          Business
dtype: object

Source: stackoverflow.com

Similar codes

2 votes188 views2 answers
Programm the fibonacci sequence
Python
0 vote176 views3 answers
Pydrive download by url
Python
1 vote180 views2 answers
How to define the name of your tkinter window
Python
0 vote245 views1 answer
Day 27: Testing hackerrank solution python
Python
2 votes167 views1 answer
Father's day 2020
Python
13 votes258 views2 answers
Delete database entry using name django
Python Django
0 vote175 views1 answer
My Skile
Python
0 vote224 views1 answer
How to make booking website using django step by step
Python Django
55 votes200 views5 answers
How to clear the console python
Python
0 vote180 views1 answer
Get rid of axes numbers matplotlib
Python