Question

我需要你在以下情况下的专业知识。

我有一个这样的集合：

"array" : {
    "item" : 1,
    "1" : [100, 130, 255],
}

"array" : {
    "item" : 2,
    "1" " [0, 70, 120],
}

"array" : {
    "item" : 3,
    "1" : [100, 90, 140],

}

我正在查询这个集合：

 db.test.find(array.1 : {$in : [100, 80, 140]});

这会返回项目编号1和3，因为它将提供的数组中的任何值与集合中的值匹配。但是，我想对此数组进行排序，以便为我提供更相似数字的结果。结果应分别为第3项和第1项。

然而，我可以获取结果并使用k近邻算法对数组进行排序。然而，处理庞大的数据集会使这非常不受欢迎（或者是吗？） MongoDB中是否有任何功能可以提供此功能？我使用Java，任何算法都足够快地实现这个目标吗？任何帮助表示赞赏。

感谢。

Answer 1

您可以使用聚合框架执行此操作，但这并不容易。麻烦在于没有$in运算符作为聚合框架的一部分。因此，您必须以编程方式匹配数组中的每个项目，这会非常混乱。编辑：重新排序，以便首先匹配，以防$in帮助您过滤掉好的部分。

db.test.aggregate(
  {$match:{"array.1":{$in:[100, 140,80]}}}, // filter to the ones that match
  {$unwind:"$array.1"}, // unwinds the array so we can match the items individually
  {$group: { // groups the array back, but adds a count for the number of matches
    _id:"$_id", 
    matches:{
      $sum:{
        $cond:[
          {$eq:["$array.1", 100]}, 
          1, 
          {$cond:[
            {$eq:["$array.1", 140]}, 
            1, 
            {$cond:[
              {$eq:["$array.1", 80]}, 
              1, 
              0
              ]
            }
            ]
          }
          ]
        }
      }, 
    item:{$first:"$array.item"}, 
    "1":{$push:"$array.1"}
    }
  }, 
  {$sort:{matches:-1}}, // sorts by the number of matches descending
  {$project:{matches:1, array:{item:"$item", 1:"$1"}}} // rebuilds the original structure
);

输出：

{
"result" : [
    {
        "_id" : ObjectId("50614c02162d92b4fbfa4448"),
        "matches" : 2,
        "array" : {
            "item" : 3,
            "1" : [
                100,
                90,
                140
            ]
        }
    },
    {
        "_id" : ObjectId("50614bb2162d92b4fbfa4446"),
        "matches" : 1,
        "array" : {
            "item" : 1,
            "1" : [
                100,
                130,
                255
            ]
        }
    }
],
"ok" : 1
}

如果您将matches字段排除在最后$project之外，则可以将{{1}}字段退出。

查询MongoDB数组并使用最匹配的元素进行排序

1 个答案: