如何在Django queryset中实现带映射规则的条件求和

你可以直接通过Django ORM内置的条件表达式+聚合函数，把映射计算逻辑下推到数据库层完成，完全不需要在Python层循环遍历查询集，性能远高于逐行加载内存计算的写法。

核心实现方式 #

用Case+When构造条件分支，对应SQL层面的CASE WHEN逻辑，再套Sum聚合函数直接拿到计算结果：

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

# 固定映射写法
total = A.objects.aggregate(
    result=Sum(
        Case(
            When(type='x', then=1),
            When(type='y', then=25),
            When(type='z', then=-3),
            output_field=IntegerField()
        )
    )
)['result']
print(total)

如果你要复用已有的映射字典，不想硬写分支，可以动态构造When子句：

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

mapp = {'x': 1, 'y': 25, 'z': -3}
# 动态生成所有条件分支
when_list = [When(type=type_key, then=type_value) for type_key, type_value in mapp.items()]

total = A.objects.aggregate(
    result=Sum(
        Case(*when_list, output_field=IntegerField())
    )
)['result']
print(total)

补充说明 #

• 这种写法所有计算逻辑都会被翻译成原生SQL在数据库侧执行，最终只返回一个求和后的数值，不会把全量数据加载到Python内存，数据量越大性能优势越明显。
• 如果type字段可能出现映射字典里不存在的值，可以给Case加上default参数指定默认值，比如default=0就会把未知类型按0计算，避免出现NULL值被Sum忽略导致结果不准的问题。
• 按照你给出的示例数据计算，最终返回的结果是1*9 + 25*2 + (-3)*1 = 56，和你原有循环写法的计算结果完全一致。

内容的提问来源于stack exchange，提问作者Maifee Ul Asad