#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. Теперь я понимаю, что вы имели в виду. Вы хотели, чтобы новый столбец содержал имена столбца, содержащего наибольшее значение в этой строке. Понял.