Функция Python, которая выводит имя столбца, а не значение

#python #pandas #numpy

#python #pandas #numpy

Вопрос:

Есть ли способ заставить python выводить имя столбцов во фрейме данных, а не значение. Я хочу, чтобы функция создавала новый столбец, в котором указано имя самого высокого столбца для каждой строки. Аналогично тому, что у меня есть ниже, но вместо вывода значения я бы хотел, чтобы вместо этого выводилось имя столбца.

 df['name'] = np.max(df, axis = 1))
  

Заранее спасибо

Комментарии:

1. Попробуйте df.idxmax(axis=1)

2. не df.columns работает?

3. @ShubhamSharma отлично сработало, спасибо!

4. Теперь я понимаю, что вы имели в виду. Вы хотели, чтобы новый столбец содержал имена столбца, содержащего наибольшее значение в этой строке. Понял.