Django查询：按ID列表过滤后如何为每个ID获取5条限定结果

要实现每个匹配的answer_id各获取5条Comment数据，不能直接在全局QuerySet后加切片，因为那会限制总结果数。以下是两种可行的实现方式：

方法一：循环查询每个Answer的评论（简单直观） #

先获取目标Answer的ID列表，再逐个查询每个Answer对应的前5条评论，最后合并结果：

# 第一步：获取目标Answer的ID列表
ans_ids = Answer.objects.filter(question=1).values_list('id', flat=True)

# 第二步：初始化空QuerySet，逐个合并每个Answer的前5条评论
from django.db.models import QuerySet

all_comments = QuerySet(model=Comment)
for ans_id in ans_ids:
    # 取出当前Answer的前5条评论
    ans_comments = Comment.objects.filter(answer_id=ans_id)[:5]
    # 合并到总结果（union会自动去重，若不需要去重可改用itertools.chain拼接）
    all_comments = all_comments.union(ans_comments)

方法二：使用Window函数（高效单查询） #

利用Django的Window函数按answer_id分组，给每个分组内的评论编号，再筛选出每个分组的前5条，只需一次数据库查询：

from django.db.models import Window, F
from django.db.models.functions import RowNumber

# 先获取目标Answer的ID列表
ans_ids = Answer.objects.filter(question=1).values_list('id', flat=True)

# 用Window函数给每个answer_id下的评论编号，筛选前5条
comments = Comment.objects.filter(answer_id__in=ans_ids).annotate(
    # 按answer_id分区，按评论ID升序（可替换为你需要的排序字段，如创建时间）
    row_num=Window(
        expression=RowNumber(),
        partition_by=F('answer_id'),
        order_by=F('id').asc()
    )
).filter(row_num__lte=5)

原代码问题说明 #

你之前的Comment.objects.filter(answer_id__in=ans.values('id'))[0:5]是对所有符合条件的评论整体做切片，所以只会返回前5条结果，不会按answer_id分组限制数量。

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