Группировать по и суммировать с использованием подчеркивания/Lodash

StackOverflow https://stackoverflow.com//questions/25047463

Вопрос

У меня есть JSON вот так:

[
  {
     platformId: 1,
     payout: 15,
     numOfPeople: 4
  },
  {
     platformId: 1,
     payout: 12,
     numOfPeople: 3

  },
  {
     platformId: 2,
     payout: 6,
     numOfPeople: 5

  },
  {
     platformId: 2,
     payout: 10,
     numOfPeople: 1
  },

]

И я хочу сгруппировать его по platformId с суммой payout и numOfPeople.

Т.е.в результате я хочу JSON вот так:

[
  "1": {
     payout: 27,
     numOfPeople: 7
   },

  "2": {
     payout: 16,
     numOfPeople: 6
  }
] 

Я пытался использовать underscore.js's _.groupBy метод, и он хорошо группируется, но как я могу получить СУММУ значений свойств объектов, как я продемонстрировал выше?

Это было полезно?

Решение

Вы можете сделать это без подчеркивания:

var result = data.reduce(function(acc, x) {
  var id = acc[x.platformId]
  if (id) {
    id.payout += x.payout
    id.numOfPeople += x.numOfPeople
  } else {
    acc[x.platformId] = x
    delete x.platformId
  }
  return acc
},{})

Но зачем вам объект с цифровыми клавишами?Вы можете преобразовать его обратно в коллекцию:

var toCollection = function(obj) {
  return Object.keys(obj)
    .sort(function(x, y){return +x - +y})
    .map(function(k){return obj[k]})
}

toCollection(result)

Обратите внимание, что объекты мутируют, поэтому вы можете сначала клонировать их, если хотите сохранить исходные данные.

Другие советы

Вот Лодаш решение такого рода проблемы.Это похоже на Underscore, но с некоторыми более продвинутыми функциями.

const data = [{
    platformId: 1,
    payout: 15,
    numOfPeople: 4
  },
  {
    platformId: 1,
    payout: 12,
    numOfPeople: 3

  },
  {
    platformId: 2,
    payout: 6,
    numOfPeople: 5

  },
  {
    platformId: 2,
    payout: 10,
    numOfPeople: 1
  },
];

const ans = _(data)
  .groupBy('platformId')
  .map((platform, id) => ({
    platformId: id,
    payout: _.sumBy(platform, 'payout'),
    numOfPeople: _.sumBy(platform, 'numOfPeople')
  }))
  .value()

console.log(ans);
<script src="https://cdn.jsdelivr.net/lodash/4.17.4/lodash.min.js"></script>

Я попробовал сделать это функциональным способом, и результат был такой

console.log(_.chain(data)
    .groupBy("platformId")
    .map(function(value, key) {
        return [key, _.reduce(value, function(result, currentObject) {
            return {
                payout: result.payout + currentObject.payout,
                numOfPeople: result.numOfPeople + currentObject.numOfPeople
            }
        }, {
            payout: 0,
            numOfPeople: 0
        })];
    })
    .object()
    .value());

В духе лодаша one liners

_.map(_.groupBy(your_list, 'group_by'), (o,idx) => { return { id: idx, summed: _.sumBy(o,'sum_me') }})

(На мой взгляд) хороший функциональный способ использования underscore.js:

var sum = function(t, n) { return t + n; };
_.mapObject(
    _.groupBy(data, 'platformId'),
    function(values, platformId) {
        return {
            payout: _.reduce(_.pluck(values, 'payout'), sum, 0),
            numOfPeople: _.reduce(_.pluck(values, 'numOfPeople'), sum, 0)
        };
    }
);
//Input 
[{
  "platformId": 1,
  "payout": 15,
  "numOfPeople": 4
}, {
  "platformId": 1,
  "payout": 12,
  "numOfPeople": 3
}, {
  "platformId": 2,
  "payout": 6,
  "numOfPeople": 5
}, {
  "platformId": 2,
  "payout": 10,
  "numOfPeople": 1
}]


//Output
{
 "1": {
    "payout": 43,
    "numOfPeople": 13
 },
 "2": {
    "payout": 43,
    "numOfPeople": 13
 }
}


// Code
output = _.reduce(input, function(acc, val, key) {
  acc[val.platformId] = {
     payout: _.sum(a, 'payout'),
     numOfPeople: _.sum(a, 'numOfPeople')
   };
  return acc;
  }, {});
Лицензировано под: CC-BY-SA с атрибуция
Не связан с StackOverflow
scroll top