MongoDB如何查询包含指定键集合中至少N个键的记录

核心思路 #

1. 提前定义需要匹配的目标键集合，以及最低匹配数阈值N
2. 遍历每个文档，统计文档中存在的目标键数量
3. 筛选出匹配数量≥N的文档
4. 仅保留文档中匹配到的目标键输出即可

常用场景实现 #

1. MongoDB 文档查询实现 #

假设你使用的是MongoDB存储文档，可通过以下聚合语句实现需求：

// 定义匹配参数
const targetKeys = ["k1", "k2", "k3", "k4"]; // 要匹配的目标键
const minMatchCount = 3; // 最低匹配数量

// 聚合查询
db.collection.aggregate([
  // 提取文档中存在的目标键
  {
    $addFields: {
      matchedKeys: {
        $filter: {
          input: targetKeys,
          cond: { $ne: [{ $type: "$" + "$$this" }, "missing"] }
        }
      }
    }
  },
  // 过滤符合匹配数量要求的文档
  {
    $match: {
      $expr: { $gte: [{ $size: "$matchedKeys" }, minMatchCount] }
    }
  },
  // 格式化输出，仅保留匹配到的键
  {
    $replaceRoot: {
      newRoot: {
        $arrayToObject: {
          $map: {
            input: "$matchedKeys",
            in: { k: "$$this", v: "$" + "$$this" }
          }
        }
      }
    }
  }
])

执行上述语句后，返回结果和你给出的预期输出完全一致：

doc1 {k2:1, k3:1, k4:1}
doc2 {k1:1, k3:1, k4:1}
doc3 {k1:1, k2:1, k4:1}
doc4 {k1:1, k2:1, k3:1}

2. Python 处理字典列表实现 #

如果你是在本地处理内存中的字典列表，可直接用集合交集实现：

# 定义匹配参数
target_keys = {"k1", "k2", "k3", "k4"}
min_match = 3

# 测试数据
docs = [
    {"doc1": {"k2":1, "k3":1, "k4":1, "k5":1, "k6":1}},
    {"doc2": {"k1":1, "k3":1, "k4":1, "k5":1, "k7":1}},
    {"doc3": {"k1":1, "k2":1, "k4":1, "k5":1, "k8":1}},
    {"doc4": {"k1":1, "k2":1, "k3":1, "k5":1}},
    {"doc5": {"k1":1, "k2":1}}
]

# 筛选结果
result = []
for doc_item in docs:
    doc_name, doc_content = list(doc_item.items())[0]
    matched_keys = target_keys & set(doc_content.keys())
    if len(matched_keys) >= min_match:
        result.append({
            doc_name: {k: doc_content[k] for k in matched_keys}
        })

# 打印结果
for res in result:
    print(res)

内容的提问来源于stack exchange，提问作者LF-DevJourney