MongoDB聚合中如何限制数组内推送的对象数量

要限制分组后两个数组的前16个元素，不能直接用$limit（它是限制聚合结果的文档总数），而需要在$group阶段之后使用$slice操作符来截取数组。以下是修改后的查询语句及说明：

修改后的聚合查询 #

db.getCollection("students").aggregate([
  // 筛选符合条件的学生：指定班级、已选课、从未挂科
  { "$match": {
    "class_number": { "$in": [49, 50, 16] },
    "grades.curr_class.percentile": { "$exists": true },
    "grades.min": { "$gte": 80 }
  }},
  // 按当前班级成绩降序排序，确保后续push的数组顺序正确
  { "$sort": { "grades.curr_class.score": -1 } },
  // 按班级分组，分别收集高GPA和低GPA的学生ID
  { "$group": {
    "_id": "$class_number",
    "studentsWithHighGPA": {
      "$push": {
        "$cond": [{ "$gte": ["$grades.gpa", 80] }, { "id": "$_id" }, "$$REMOVE"]
      }
    },
    "studentsWithoutHighGPA": {
      "$push": {
        "$cond": [{ "$lt": ["$grades.gpa", 80] }, { "id": "$_id" }, "$$REMOVE"]
      }
    }
  }},
  // 新增阶段：截取每个数组的前16个元素
  { "$project": {
    "studentsWithHighGPA": { "$slice": ["$studentsWithHighGPA", 16] },
    "studentsWithoutHighGPA": { "$slice": ["$studentsWithoutHighGPA", 16] }
  }}
])

关键说明 #

• $limit的局限性：$limit是对整个聚合管道输出的文档数量做限制，无法作用于分组后生成的数组内部元素，因此之前的尝试无效。
• $slice的作用：在$group完成数组构建后，通过$project阶段使用$slice，可以精准截取每个数组的前N个元素（这里是16）。
• 排序的必要性：$sort阶段放在$match之后，既保证了排序的效率（先过滤再排序），又确保了分组后数组内的学生是按当前班级成绩降序排列的，因此$slice取前16个就是该班级对应分组中成绩最高的16名学生。
• $$REMOVE的作用：MongoDB 4.2+支持的$$REMOVE会自动排除不符合条件的元素，避免数组中出现null或无效值，后续$slice可以直接处理干净的数组。

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