Агрегация нескольких моделей - Django

Question

Я пытаюсь вычислить среднее значение поля по различным подмножествам набора запросов.

Player.objects.order_by('-score').filter(sex='male').aggregate(Avg('level')) 

Это прекрасно работает!

---

Но ... если я пытаюсь вычислить его для 50 лучших игроков это не работает.

Player.objects.order_by('-score').filter(sex='male')[:50].aggregate(Avg('level')) 

Этот последний возвращает тот же результат, что и запрос над ним, что неверно.

---

Что я делаю неправильно?

Помощь была бы очень признательна!

Answer 1

topfifty = Player.objects.order_by('-score')[:50] 
Player.objects.filter(sex='male',pk__in=topfifty).aggregate(avglevel=Avg('level')) 

Редактировать: я не тестировал это, но думаю, вы поняли, куда я иду.

topfifty = Player.objects.order_by('-score')[:50] 
ids = [] 
for t in topfifty: 
    ids += [t.id] 

Player.objects.filter(sex='male',pk__in=ids).aggregate(avglevel=Avg('level')) 

Это хакерское, но лучшее, что я могу придумать. возможно, рассмотреть фильтрацию для мужчин и взять 50 лучших мужчин вместо этого, который берет верхнюю пядь, а затем отфильтровывает самцов.

Answer 2

Хм. Документы говорят, что

«Нарезка. Как описано в разделе Ограничение запросов QuerySet, QuerySet можно нарезать, используя синтаксис массива Python. Обычно нарезка QuerySet возвращает другой (неоцененный) QuerySet, но Django будет выполнять запрос базы данных, если вы используете параметр «шаг» синтаксиса среза ». http://docs.djangoproject.com/en/dev/ref/models/querysets/

Так что я хотел бы попробовать

Player.objects.order_by('-score').filter(sex='male')[0:50:1].aggregate(Avg('level')) 

Answer 3

Разбейте задачу на два логических шагов; определить целевой набор, выполнить совокупный расчет.

top50_male_players = Player.objects.filter(sex='male').order_by('-score')[:50] 
result = Player.objects.filter(pk__in=top50_male_players).aggregate(Avg('level')) 

Благодаря ленивой оценке QuerySet он будет выполняться в одной операции с БД. Вы можете играть с этим в оболочке, чтобы проверить количество запросов.

> from django.db import connection 
> connection.queries = [] 
> top50_male_players = Player.objects.filter(sex='male').order_by('-score')[:50] 
> len(connection.queries) 
0 
> result = Player.objects.filter(pk__in=top50_male_players).aggregate(Avg('level')) 
> len(connection.queries) 
1 
> result 
{'level__avg': <some number>} 
> len(connection.queries) 
1