如何在Django中为每个Foo对象计算前6条FooStatement的收入平均值并注释

嘿，这个问题我之前在项目里也碰到过类似的，结合你用的Django 1.11.6和PostgreSQL，最靠谱的解法就是利用PostgreSQL的窗口函数来实现——毕竟数据量还在增长，在数据库层面处理比拉到Python里算高效太多了！

为什么直接切片行不通？ #

你之前尝试的切片操作（比如Foo.objects.annotate(...)[6:]）其实是对整个Foo查询集进行切片，而不是针对每个Foo对象关联的FooStatement单独取前6条，这肯定达不到你想要的效果。而且每个Foo的记录起始日期不同，用日期阈值筛选也不现实，窗口函数刚好能解决这个“分组取Top N”的需求。

具体实现步骤 #

1. 给FooStatement添加行号标记（按Foo分组、日期排序）

我们用PostgreSQL的ROW_NUMBER()窗口函数，给每个Foo下的FooStatement按日期排序分配行号，这样就能轻松筛选出每个Foo的前6条记录。

2. 筛选前6条记录并计算平均值

有了行号标记后，我们只需要过滤出行号≤6的记录，再按Foo分组计算收入平均值即可。

代码示例 #

方法一：用RawSQL实现高效查询

这种方式直接在数据库层面完成所有计算，性能最优：

from django.db.models import Avg, Q
from django.db.models.expressions import RawSQL

# 第一步：给每个FooStatement标记行号（按Foo分组，日期升序排序，若要最新6条则改成DESC）
top6_statements = FooStatement.objects.annotate(
    row_number=RawSQL(
        'ROW_NUMBER() OVER (PARTITION BY foo_id ORDER BY date ASC)',
        []
    )
).filter(row_number__lte=6)

# 第二步：给每个Foo计算前6条记录的收入平均值
foo_with_avg = Foo.objects.annotate(
    avg_top6_revenue=Avg(
        'foostatement__revenue',
        filter=Q(foostatement__in=top6_statements)
    )
)

方法二：子查询方式（更直观）

如果你觉得上面的方式有点绕，也可以用子查询直接计算每个Foo的前6条平均值：

from django.db.models.expressions import RawSQL

foo_with_avg = Foo.objects.annotate(
    avg_top6_revenue=RawSQL(
        '''
        SELECT AVG(revenue)
        FROM (
            SELECT revenue
            FROM app_foostatement  -- 替换成你的app名+表名
            WHERE foo_id = app_foo.id
            ORDER BY date ASC
            LIMIT 6
        ) AS subquery
        ''',
        []
    )
)

关键优化点 #

• 索引优化：给FooStatement的foo和date字段建联合索引，能大幅提升窗口函数的执行速度：

  class FooStatement(models.Model):
      foo = models.ForeignKey(Foo, on_delete=models.CASCADE)
      revenue = models.DecimalField(decimal_places=2, max_digits=10)  # 注意补充max_digits，否则Django会报错
      date = models.DateField()
  
      class Meta:
          indexes = [
              models.Index(fields=['foo', 'date']),
          ]
  

• 排序方向：如果需要的是最新的6条记录，把代码里的ORDER BY date ASC改成ORDER BY date DESC即可。
• 空值处理：如果某个Foo没有关联的FooStatement，avg_top6_revenue会返回None，你可以用Coalesce函数设置默认值（比如Coalesce(Avg(...), Value(0))）。

验证结果 #

你可以通过以下方式验证结果是否正确：

for foo in foo_with_avg[:5]:
    print(f"Foo: {foo.name}, 前6条收入平均值: {foo.avg_top6_revenue}")

内容的提问来源于stack exchange，提问作者Don