python - MongoDB、Flask、使用 $unwind 查询进行聚合

Question

上下文

我有一个使用 pymongo 连接到 MongoDB 的 Flask 应用程序。目前数据库包含以下格式的运行数据:

{
    "_id": {
        "$oid": "5df031cec687bf2b4c4349b9"
    },
    "run_number": "1",
    "frames": [{
        "frame_number": 1,
        "data": {
            "brake": "0.1",
            "steer": "0.4",
            "throttle": "0.6"
        }
    }, {
        "frame_number": 2,
        "data": {
            "brake": "0.2",
            "steer": "0.8",
            "throttle": "0.6"
        }
    }, {
        "frame_number": 3,
        "data": {
            "brake": "0.6",
            "steer": "0.2",
            "throttle": "0.1"
        }
    }]
}

我能够使用此端点检索特定运行的所有数据:

@app.route('/pitcrew-purple/api/v1/<run_number>/')
def get_run(run_number):
    if run_number:
        data = []
        for i in mongodb.pitcrewdb.find({"run_number": run_number}):
            i.pop('_id')
            data.append(i)
        if not data:
            return "No data was found for run number {}".format(run_number), 400
        return jsonify(data), 200
    return "No run_number was given", 400

问题

我正在开发一个 Flask API 端点，它将返回来自某个 run_number 的所有数据，其中frame_number 大于或等于给定的帧号。

因此，当将 run_number = 1 和 frame_number = 2 作为输入给出时，来自运行编号 1 的帧编号 >= 2 的所有数据应该被检索。

我是 MongoDB 新手，已阅读文档，但无法从数据库获取我想要的数据。

我尝试过的

我尝试使用此端点检索所需的数据

@app.route('/pitcrew-purple/api/v1/<run_number>/<start_frame_number>/')
def get_rundata_from_start_frame(run_number, start_frame_number):
    if run_number and start_frame_number:
        data = []
        query = mongodb.pitcrewdb.find(
            {"run_number": run_number, "frames.frame_number": {"$gte": start_frame_number}}
        )
        for i in query:
            i.pop("frames.frame_number")
            data.append(i)
        if not data:
            return "The specified data starting from frame number: {} was not found".format(start_frame_number), 400
        return jsonify(data), 200
    return "No parameters were given", 400

我还尝试将 i.pop("frames.frame_number") 替换为 i.pop("_id") 但这两者都会导致“指定的未找到从帧号:2 开始的数据”。

为了从 MongoDB 中检索所需的数据，我应该更改什么？

编辑预期输出

"run_number": "1",
    "frames": [{  
      {
        "frame_number": 2,
        "data": {
            "brake": "0.2",
            "steer": "0.8",
            "throttle": "0.6"
        }
    }, {
        "frame_number": 3,
        "data": {
            "brake": "0.6",
            "steer": "0.2",
            "throttle": "0.1"
        }
    }]

Answer 1

您可以通过使用 $unwind 和 $group 以及 $push 来实现这一点，如下所示:

db.pitcrewdb.aggregate([
  {
    "$match": {
      "run_number": "1",
      "frames.frame_number": {
        "$gte": 2
      }
    }
  },
  {
    "$project": {
      "_id": 0
    }
  },
  {
    "$unwind": "$frames"
  },
  {
    "$match": {
      "frames.frame_number": {
        "$gte": 2
      }
    }
  },
  {
    "$group": {
      "_id": "$run_number",
      "frames": {
        "$push": "$frames"
      }
    }
  }
])

$unwind 将破坏您的数组，然后您过滤结果，并在 $group 之后再次过滤

或者您也可以使用 $filter

db.pitcrewdb.aggregate([
  {
    "$match": {
      "run_number": "1"
    }
  },
  {
    "$project": {
      "_id": 0,
      "frames": {
        $filter: {
          input: "$frames",
          as: "frame",
          cond: {
            $gte: [
              "$$frame.frame_number",
              2
            ]
          }
        }
      }
    }
  }
])