Найти общее количество созданного пользователем контента, для которого создано более 500 единиц контента, Python

Question

Вот .head() данных, которые я использую для этого расчета:

Теперь я уже сделал это:

group = user_message.groupby('user_id')
agg = group.aggregate({'content_count': np.sum})
agg

, который дает мне это для вывода:

Теперь я хочу просто вернуть user_id, которые имеют content_count больше 500. Я подумал, что могу просто изменить эту строку: agg = group.aggregate({'content_count': np.sum > 500}) Но я получаю эту ошибку при этом:

TypeError: '>' not supported between instances of 'function' and 'int'

Answer 1

Попробуйте

agg[agg.content_count>500]