Select 每个数组中的前 n 项按 desc 排序

Select top n items from each array order by desc

考虑包含以下文档的集合:

{
    "_id" : 1,
    "name" : "Class 1",
    "students" : [  
        { "rollNo" : 10001, "name" : "Ram", "score" : 65 },
        { "rollNo" : 10002, "name" : "Shyam", "score" : 90 }, <=
        { "rollNo" : 10003, "name" : "Mohan", "score" : 75 }  <=
    ]
},
{
    "_id" : 2,
    "name" : "Class 2",
    "students" : [  
        { "rollNo" : 20001, "name" : "Krishna", "score" : 88 },
        { "rollNo" : 20002, "name" : "Sohan", "score" : 91 }, <=
        { "rollNo" : 20003, "name" : "Radhika", "score" : 82 },
        { "rollNo" : 20004, "name" : "Komal", "score" : 55 },
        { "rollNo" : 20005, "name" : "Sonam", "score" : 91 }  <=        
    ]
},
{
    "_id" : 3,
    "name" : "Class 3",
    "students" : [  
        { "rollNo" : 30001, "name" : "Monika", "score" : 77 }, <=
        { "rollNo" : 30002, "name" : "Rahul", "score" : 81 }   <=
    ]
}

我的 objective 是从每个数组中按分数降序排列前 N 个学生(考虑按分数排序的前 2 个学生)。

我的预期结果是:

如果您只想要没有分组的学生记录,您可以简单地 $unwind 然后 $sort:

db.collection.aggregate([
  { $unwind: "$students" },
  { $sort: { "students.score": -1 } },
  { $limit: 2 }
])

那将留下 students 对象。为了获得更清晰的输出,您可以使用 $replaceRoot$mergeObjects:

db.collection.aggregate([
  { $unwind: "$students" },
  { $sort: { "students.score": -1 } },
  { $replaceRoot: {
      newRoot: {
        $mergeObjects: [ { _id: "$_id", class: "$name" }, "$students" ]
      }
    }
  },
  { $limit: 2 }
])

看到了working here

这将为您提供以下输出:

[
  {
    "_id": 2,
    "class": "Class 2",
    "name": "Sonam",
    "rollNo": 20005,
    "score": 91
  },
  {
    "_id": 2,
    "class": "Class 2",
    "name": "Sohan",
    "rollNo": 20002,
    "score": 91
  }
]

更新:

使用此方法获得每个组的前 2 名:

db.collection.aggregate([
  { $unwind: "$students" },
  { $sort: { "students.score": -1 }
  },
  {
    $group: {
      "_id": "$_id",
      "name": { $first: "$name" },
      "students": { $push: "$students" }
    }
  },
  {
    "$project": {
      "top_two": { "$slice": [ "$students", 2 ] }
    }
  }
])

看到了working here

Mongo 5.2 开始,这是新 $sortArray and $firstN 聚合运算符的一个很好的用例:

// { students: [
//    { name: "Ram",   score: 65 },
//    { name: "Shyam", score: 90 },
//    { name: "Mohan", score: 75 }
// ]}
// { students: [
//    { name: "Krishna", score: 88 },
//    { name: "Sohan",   score: 91 },
//    { name: "Radhika", score: 82 },
//    { name: "Komal",   score: 55 },
//    { name: "Sonam",   score: 91 }
// ]}
// { students: [
//    { name: "Monika", score: 77 },
//    { name: "Rahul",  score: 81 }
// ]}
db.collection.aggregate(
  { $set: { students: {
    $firstN: {
      n: 2,
      input: { $sortArray: { input: "$students", sortBy: { score: -1 } } }
    }
  }}}
)
// { students: [
//    { name: "Shyam", score: 90 },
//    { name: "Mohan", score: 75 }
// ]}
// { students: [
//    { name: "Sohan", score: 91 },
//    { name: "Sonam", score: 91 }
// ]}
// { students: [
//    { name: "Rahul", score: 81 },
//    { name: "Monika", score: 77 }
// ]}

这个:

  • score
  • 对每个 students 数组进行排序 ($sortArray)
  • 为了获得前 2 ($firstN) 名学生