Что выведет код Junior groupby и агрегаты спрашивают часто обновлено

groupby: size против count

Вопрос

Что выведет код? По строке на каждый print.

import pandas as pd

df = pd.DataFrame({
    'city': ['A', 'A', 'B', 'B', 'B'],
    'amount': [10, None, 5, None, None],
})
g = df.groupby('city')
print(g.size().tolist())
print(g['amount'].count().tolist())
print(g['amount'].sum().tolist())

Попробуйте сами

Напишите, что выведет код: каждый вывод с новой строки.

Показать ответ и разбор

Коротко

Код выведет: [2, 3], [1, 1], [10.0, 5.0]. size считает строки: в A две, в B три. count считает непустые значения: по одному. sum пропускает NaN, и так как в колонке есть NaN, она имеет тип float, отсюда 10.0 и 5.0.

Что выведет

[2, 3]
[1, 1]
[10.0, 5.0]

Разбор

size считает строки: в A две, в B три. count считает непустые значения: по одному. sum пропускает NaN, и так как в колонке есть NaN, она имеет тип float, отсюда 10.0 и 5.0.

Ответьте сами, а не прочитайте

В тренажёре вопрос попадёт в повторения и вернётся, когда начнёт забываться. Плюс билеты с зачётом: 20 вопросов, не больше 2 ошибок.

Термины по теме