Реализация кучи с использованием бинарного дерева

Когда уже заданные сообщения отвечают на вопрос, я сделал небольшое изменение, добавив имя столбца, для которого применяется функция max () для лучшей читаемости кода.

df.groupby('A', as_index=False)['B'].max()