Ваша ошибка вызвана [0]
в конце вашей строки, где вы делаете groupby
. Вы не опубликовали полное сообщение об ошибке, но держу пари, что у вас есть KeyError: 0
. Это связано с тем, что в вашем индексе больше нет 0
. Если вы посмотрите на DataFrame
, созданный после groupby
, вы увидите, что теперь у вас есть иерархический индекс, созданный из уникальных комбинаций значений column1
и column2
.
Быстрое решение? Замените [0]
на .iloc[0]
, чтобы получить строку в месте нулевого индекса.
output = df.groupby(['column1','column2']).count().sort_values(by=['column1','column2'], axis = 0).iloc[0]
Или используйте .head(1)
, чтобы получить верхний ряд DataFrame
.