как найти расположение столбца в сводной таблице - PullRequest
1 голос
/ 04 ноября 2019

Привет, у меня есть сводная таблица с использованием метода pivot_table ().

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

этот сводный_файл

EventID 1.0 2.0 3.0 4.0 5.0 6.0
Name                        
John    10  90  0   70  30  50
Berry   20  50  30  0   0   0
Charles 50  20  0   80  40  60
Susan   60  30  30  0   30  0
Elisa   200 30  30  100 0   0

Ожидаемый вывод:

Give me the eventID for highest amount of Charles = 4.0
Give me the eventID for highest amount of John = 2.0

Я пробовал метод .loc, но только если я знаюEventID. У меня есть только имя и сумма

pivot_df.loc(Berry,3.0)  = 30

1 Ответ

1 голос
/ 04 ноября 2019

Это .loc и idxmax

df.loc['Charles'].idxmax()

Out[151]: 4.0

df.loc['John'].idxmax()

Out[150]: 2.0

Если вы хотите получить оба одновременно

df.idxmax(1).loc[['Charles', 'John']]

Out[153]:
Charles    4.0
John       2.0
dtype: float64
...