How to sum arrays from different documents in MongoDB Aggregation?

Question

I have a collection of documents, each containing an array of revenues of different lengths.
I want to get a single array of revenue from the values that match the query.

Example data

...
{"cohort": "2112", "revenue": [1, 1, 0, 0, 5], ...},
{"cohort": "2113", "revenue": [0, 0, 2, 0], ...},
{"cohort": "2114", "revenue": [0, 1, 3], ...}
...

Expected result for cohorts 2113 and 2114

[0, 1, 5] or [0, 1, 5, 0]

The two results are equal for my purpose, since I know the length of the shortest array.
Is there any way to perform the operation with MongoDB Aggregate pipeline?
Or can you suggest a better solution?

And yes, I use PyMongo to access the database.

dgiugg · Accepted Answer · 2016-04-28 13:57:54Z

0

I just discovered this new 3.2 feature: includeArrayIndex.

So you can unwind the field revenue with this option and then sum using your includeArrayIndex field as id and then sort and push to get a new array.

Reference: https://docs.mongodb.org/manual/reference/operator/aggregation/unwind/

answered Apr 28, 2016 at 13:57

dgiugg

1,33414 silver badges24 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user3043365 Over a year ago

Thanks! I missed that includeArrayIndex option

Collectives™ on Stack Overflow

How to sum arrays from different documents in MongoDB Aggregation?

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related