python - 如何计算 Django 中跨字段的值？

Question

模型

class ModelA(models.Model):
    name = models.CharField()

class ModelB(models.Model):
    MY_CHOICES = (
        ('X', 'X'),
        ('Y', 'Y'),
        ('Z', 'Z'),
    )
    modela = models.ForeignKey(ModelA, on_delete=models.CASCADE)
    txt_1 = models.CharField(choices=MY_CHOICES)
    txt_2 = models.CharField(choices=MY_CHOICES)

鉴于上面的简化示例，如果有两个字段需要计算，我如何计算每个选项值被记录了多少次？

理想情况下，结果应该是这样的:

{'X': 15, 'Y': 27, 'Z': 89}

我尝试了以下方法，但在我的真实模型中，我有大约 20 个字段要计算，这并没有给出我希望的结果:

ModelA.objects.values('modelb__txt1', 'modelb__txt2').annotate(Count('modelb__txt1', 'modelb__txt2'))

我之前创建了巨大的字典并手动对值进行排序/计数，但现在这是难以管理且丑陋的。

Answer 1

使用一个查询(对于有限数量的列)

通过一个查询，我们可以这样做:

from django.db.models import Count

qs = ModelB.objects.values('txt_1', 'txt_2').annotate(
    cnt=Count('id')
).order_by('txt_1', 'txt_2')

但现在我们仍然没有，因为现在我们有 txt_1 和 txt_2 的每个组合的元素数量。我们想将其“扁平化”到每个单独的选择。例如，我们可以通过构造一个 Counter [Python-doc] 来做到这一点。 :

from collections import Counter

result = Counter()
for row in qs:
    result[row['txt_1']] += row['cnt']
    result[row['txt_2']] += row['cnt']

对于此 QuerySet 的每一行，我们因此将数字 (cnt) 添加到两个键。因此，这意味着我们对 txt_1 和 txt_2 都具有值 'X' 两次的行进行计数。

Counter 是字典的子类，但如果您想将其转换为 dictionary，您可以稍后编写:

result_dict = dict(result)

从未选择的选项将不会出现在字典中，因为查询集不会包含这些，因此我们永远不会将这些添加到 Counter。但是我们当然可以对字典进行后处理，并为这些添加0。

n 个查询(n 个列数)

以上通常会很好地工作。但是，如果选择的数量很大，则处理将更多地在 Python 端进行，这通常较慢。然后我们可以进行线性化，并处理两个查询:

from collections import Counter
from django.db.models import Count

result = Counter()
for col in ['txt_1', 'txt_2']:
    qs = ModelB.objects.values(col).annotate(cnt=Count('id')).order_by(col)
    result.update({q[col]: q['cnt'] for q in qs})

这将减少两个查询。但在这种情况下，每个查询将(最多)返回三行。而另一种方法将导致一个查询返回(最多)九行。对于少量行，这不是问题。但是案例的数量很容易随列数呈指数增长。