Как посчитать среднее значение данных в массиве?

Question

[Gauthier]

Есть массив:

[
    {
        "properties": {
            "name": "Название 1",
            "id": 1,
            "groups": [
                {
                    "id": 0,
                    "name": "Группа 1",
                    'well-being': 0.9
                },
                {
                    "id": 1,
                    "name": "Группа 2",
                    'well-being': 0.5
                },
                {
                    "id": 2,
                    "name": "Группа 3",
                    'well-being': 0.4
                }
            ]
        }
    },
    {
        "properties": {
            "name": "Название 2",
            "id": 2,
            "groups": [
                {
                    "id": 0,
                    "name": "Группа 1",
                    'well-being': 0.8
                },
                {
                    "id": 1,
                    "name": "Группа 2",
                    'well-being': 0.3
                },
                {
                    "id": 2,
                    "name": "Группа 3",
                    'well-being': 0.8
                }
            ]
        }
    },
    {
        "properties": {
            "name": "Название 3",
            "id": 3,
            "groups": [
                {
                    "id": 0,
                    "name": "Группа 1",
                    'well-being': 0.8
                },
                {
                    "id": 1,
                    "name": "Группа 2",
                    'well-being': 0.1
                },
                {
                    "id": 2,
                    "name": "Группа 3",
                    'well-being': 0.6
                }
            ]
        }
    },
]

Необходимо посчитать среднее значение well-being для каждой группы.
Например для группы 1 это значение будет 0.8, для группы 2 - 0.3 и т.д.

Предполагаю, что первоначально массив нужно прогнать через forEach, но что с ним делать дальше не понимаю.

Answer 1

[Сергей Соколов]

Сначала нужно получить объект, где ключи - id групп, а значения - массивы со всеми значениями.
Потом посчитать средние.

const data = arr.reduce((acc, c) => {
  c.properties.groups.forEach((group) => {
    if (! acc.hasOwnProperty(group.id)) acc[group.id] = [];
    acc[group.id].push(group["well-being"]);
  });
  return acc;
}, {});
/*
{
  0: [0.9, 0.8, 0.8],
  1: [0.5, 0.3, 0.1],
  2: [0.4, 0.8, 0.6]
}  */


const avg = {};
for (let id in data) {
  avg[id] = data[id].reduce((acc, c) => acc + c) / data[id].length;
}

console.log(avg); /*
{
  0: 0.8333333333333334,
  1: 0.3,
  2: 0.6000000000000001  // 0.1 + 0.2 !== 0.3
} */

Comments

[WbICHA]

for (id in data) {

Форин? Ты пьян?)

[Сергей Соколов]

WbICHA, wtf?

Мало ли, какие там у них могут быть id в этом JSON'е: 1, 2, 7, 974, "unpublished". Посему свойства объекта и for-in. Эпатаж! Выпад! Вызов! Смелый опус.

[WbICHA]

Сергей Соколов, форин, ещё и с объявлением глобального ид через вар, ты чего?) Object.keys же.

[Сергей Соколов]

WbICHA, ну-ка, расскажи, чем так лучше Object.keys чем for in. Для чего нам тут лишний массив делать?

p.s. let добавил, спс

[Дмитрий Беляев]

Сергей Соколов, например тем, что Object.keys не лезет в прототип и отдает только собственные свойства?
Хотя зачем здесь объект и итерация по нему - это более непонятно...

[Сергей Соколов]

Дмитрий Беляев, прототипы в данном примере нерелевантны. Объект тут как словарь.

[WbICHA]

Сергей Соколов, но зачем оставлять место для потенциальных ошибок? И в целом использовать антипаттерн. Тем более, что форин заменяется на Object.keys().reduce.

[Дмитрий Беляев]

Aleksandr-JS-Developer, как будто в C#/Java/Rust/etc. как то по другому?

Answer 2

[Дмитрий Беляев]

const averages = Array.from(
    arr
        .flatMap(v => v.properties.groups)
        .reduce((acc, {id, 'well-being': value}) => {
            const [count, total] = acc.get(id) ?? [0, 0];
            acc.set(id, [count + 1, total + value]);
            return acc;
        }, new Map())
        .entries(),
    ([id, [count, total]]) => ({id, average: total / count}),
);
console.log(averages);