groupByToMap would be useful in that case as pointed out in another comment. Updated the post with that.