如何在 Django 查询集中对条件求和

Question

我正在尝试将 Django 查询与条件相加。 假设我有一些这样的数据：

| Name | Type |
---------------
| a    | x    |
| b    | z    |
| c    | x    |
| d    | x    |
| e    | y    |
| f    | x    |
| g    | x    |
| h    | y    |
| i    | x    |
| j    | x    |
| k    | x    |
| l    | x    |

这些类型是字符串，它们有值，比如x = 1, y = 25, z = -3

如何在没有循环的情况下总结所有值？ 目前使用循环。

data = A.objects.all()
sum = 0
mapp = {'x': 1, 'y': 25, 'z': -3}
for datum in list(data):
    sum = sum + mapp[datum.type]
print(sum)

Answer 1

要在数据库中执行计算，请使用Queryset.aggregate()和使用Case / When的聚合表达式：

from django.db.models import Sum, Case, When

A.objects.all().aggregate(
    amount=Sum(Case(
        When(type="x", then=1), 
        When(type="y", then=25), 
        When(type="z", then=-3),
        default=0,
        output_field=FloatField()
    ))
)

Answer 2

您可以将其简写为：

sum(mapp.get(o['Type'],0) for o in data)

如果您相信数据具有所有有效类型，则更简单：

sum(mapp[o['Type']] for o in data)

（尽管不要相信数据）

Answer 3

如果映射x = 1, y = 25等...来自另一个表，您可以使用一些 SQL voodoo 让数据库为您处理求和。 否则，您必须（以一种或另一种方式）遍历所有结果并将它们总结起来。

理论上，您也可以只计算表中x 、 y和z的不同数量，并根据示例的结构使sum = x*count_of_x + y*count_of_y + z*count_of_z 。