Django聚合仅计数真值

Question

我用聚合来计算一列布尔数。我想要真实值的数目。

DJANGO代码：

count = Model.objects.filter(id=pk).aggregate(bool_col=Count('my_bool_col')

返回所有行的计数。

SQL查询应为：

SELECT count(CASE WHEN my_bool_col THEN 1 ELSE null END) FROM <table_name>

这是我的实际代码：

stats = Team.objects.filter(id=team.id).aggregate(
    goals=Sum('statistics__goals'),
    assists=Sum('statistics__assists'),
    min_penalty=Sum('statistics__minutes_of_penalty'),
    balance=Sum('statistics__balance'),
    gwg=Count('statistics__gwg'),
    gk_goals_avg=Sum('statistics__gk_goals_avg'),
    gk_shutout=Count('statistics__gk_shutout'),
    points=Sum('statistics__points'),
)

感谢使用django-aggregate-if的建议

以下是解决方案：

from django.db.models import Sum
from django.db.models import Q
from aggregate_if import Count

stats = Team.objects.filter(id=team.id).aggregate(
    goals=Sum('statistics__goals'),
    assists=Sum('statistics__assists'),
    balance=Sum('statistics__balance'),
    min_penalty=Sum('statistics__minutes_of_penalty'),
    gwg=Count('statistics__gwg', only=Q(statistics__gwg=True)),
    gk_goals_avg=Sum('statistics__gk_goals_avg'),
    gk_shutout=Count('statistics__gk_shutout', only=Q(statistics__gk_shutout=True)),
    points=Sum('statistics__points'),
)

Answer 1

为Django 1.10更新。现在可以执行条件聚合：

from django.db.models import Count, Case, When
query_set.aggregate(bool_col=Count(Case(When(my_bool_col=True, then=1))))

更多信息请访问：

• https://docs.djangoproject.com/en/1.11/ref/models/conditional-expressions/#case

Answer 2

布尔伯爵的另一个解决方案是：

from django.db.models import Sum, IntegerField
from django.db.models.functions import Cast

Model.objects.filter(id=pk).annotate(bool_col=Sum(Cast('my_bool_col', IntegerField())))

只需将False转换为0和True转换为1，然后只需Sum

Answer 3

更新：

从Django 1.10开始，您可以：

from django.db.models import Count, Case, When
query_set.aggregate(
    bool_col=Count(
        Case(When(my_bool_col=True, then=Value(1)))
    )
)

阅读Conditional Expression classes

老答案

似乎您要做的是某种“条件聚合”。现在Aggregation函数不支持像filter或exclude这样的查找：fieldname，fieldname。。。

所以你可以试试这个：

django-aggregate-if

描述摘自官方页面。

Conditional aggregates for Django queries, just like the famous SumIf and CountIf in Excel.

你也可以首先为每个团队计算期望值，我的意思是为每个团队计算你感兴趣领域中的True数量。然后做所有你想做的事。