Вернуть все имена столбцов в случае связанного наибольшего значения в пандах

Question

У меня есть фрейм данных.

>df
 A    B    C    D    E
 0.1  0.8  0.3  0.8  0.3

Я хочу получить имена столбцов с наибольшим значением. Однако, если бы я использовал

maxcols = df.idxmax(axis=1)

, это дало бы только первый столбец (B) в этом случае, в то время как я хочу вывод, что-то вроде [B, D].

Answer 1

В вашем случае вы можете попробовать

s=df.eq(df.max(1),axis=0)
s[s].stack().index.get_level_values(1)
Index(['B', 'D'], dtype='object')