使用 MongoDB 压缩数组

Zip arrays with MongoDB

是否可以在 mongo 文档中压缩数组?我的意思是 zip 的函数式编程定义,其中相应的项目配对成元组。

更准确地说,我想从这样的 mongo 文件开始:

{
    "A" : ["A1", "A2", "A3"],
    "B" : ["B1", "B2", "B3"],
    "C" : [100.0, 200.0, 300.0]
}

并最终得到 mongo 个这样的文档:

{"A":"A1","B":"B1","C":100.0},
{"A":"A2","B":"B2","C":200.0},
{"A":"A3","B":"B3","C":300.0},

理想情况下,这将使用聚合框架,我已经在使用它来将我的文档带到这个阶段。

从 MongoDB 3.4 开始,我们可以使用 $zip 运算符来压缩我们的数组。

也就是说,除非您知道数组的长度,否则我们无法获得您预期的输出。

db.collection.aggregate( [ 
    { "$project": { 
        "zipped": { 
            "$zip": { "inputs": [ "$A", "$B", "$C" ] } 
        } 
    }}
])

产生:

{ 
    "_id" : ObjectId("578f35fb6db61a299a383c5b"),
    "zipped" : [
        [ "A1", "B1", 100 ],
        [ "A2", "B2", 200 ],
        [ "A3", "B3", 300 ]
    ]
}

如果我们恰好知道每个子数组的元素个数,我们可以使用$map变量运算符将return一个数组的子文件。

$map表达式中,我们需要使用$arrayElemAt运算符来设置我们字段A、B、C的值。

db.collection.aggregate( [ 
    { "$project": { 
        "zipped": { 
            "$map": { 
                "input": { 
                    "$zip": { "inputs": [ "$A", "$B", "$C" ] } 
                }, 
                "as": "el", 
                "in": { 
                    "A": { "$arrayElemAt": [ "$$el", 0 ] }, 
                    "B": { "$arrayElemAt": [ "$$el", 1 ] }, 
                    "C": { "$arrayElemAt": [ "$$el", 2 ] } 
                } 
            }
        }
    }}
] )

产生:

{
    "_id" : ObjectId("578f35fb6db61a299a383c5b"),
    "zipped" : [
        {
            "A" : "A1",
            "B" : "B1",
            "C" : 100
        },
        {
            "A" : "A2",
            "B" : "B2",
            "C" : 200
        },
        {
            "A" : "A3",
            "B" : "B3",
            "C" : 300
        }
    ]
}