Получить строки с максимальными значениями в кадре данных Pandas

Question

Выполнение функции для получения строки с максимальным значением в Pandas Dataframe, выглядит следующим образом:

def max_row(df, col):

    max_index = df[col].idxmax()

    return df.loc[[max_index]]

Так что это работает так: max_row(test_df, 'Col_A')

То, что я пытаюсь сделать, эточтобы показать так 2 строки Dataframe.В настоящее время используется функция nlargest(), но кажется, что что-то неправильно на стороне loc:

def max_rows(df, col, n):

    max_indexes = df[col].nlargest(n)
    max_indexes = list(max_indexes.index)

    return df.loc[[max_indexes]]

Как это можно написать для выполнения задачи?

Спасибо!

Answer 1

Изменить последнюю строку на

def max_rows(df, col, n):

    max_indexes = df[col].nlargest(n)
    max_indexes = list(max_indexes.index)

    return df.loc[max_indexes]