[판다스] 최댓값을 포함하는 컬럼명을 찾는 방법

문제 예시

아래와 같이 수치로 표현된 컬럼들이 있을 때, row 별로 최댓값을 가지는 컬럼명을 모아 새로운 컬럼(Max)을 생성하고자 함

Communications and Search   Business    General Lifestyle  Max
0   0.745763    0.050847    0.118644    0.084746           Communications 
0   0.333333    0.000000    0.583333    0.083333           Business  
0   0.617021    0.042553    0.297872    0.042553           Communications 
0   0.435897    0.000000    0.410256    0.153846           Communications 
0   0.358974    0.076923    0.410256    0.153846           Business

판다스 메서드 활용하여 해결

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

 

출처: https://stackoverflow.com/questions/29919306/find-the-column-name-which-has-the-maximum-value-for-each-row