Как двумерный массив порезать на куски и превратить строки в столбцы?

Question

[Alex10214]

Есть такой массив:

const t = [
  ['1Y', '3Y', '5Y', 'Since inception'],
  ['2.3', '36.57', '27.56', '24.36'],
  ['41.56', '-3.01', '2.24', '-6.12'],
  ['-11.28', '-18.78', '-18.78', '-22.7'],
  ['-8.99', '-57.64', '-57.64', '-89.62'],
  ['17.48', '23.41', '19.9', '20.83'],
  ['19.96', '24.25', '21.25', '23.72'],
  ['0.13', '1.56', '1.39', '1.17'],
  ['2.08', '-0.12', '0.11', '-0.26'],
  ];

Из него надо сделать такой массив:

const res = [
  [
    ['1Y', '2.3', '41.56'],
    ['3Y', '36.57', '-3.01'],
    ['5Y', '27.56', '2.24'],
    ['Since inception', '24.36', '-6.12'],
  ],
  [
    ['1Y', '-11.28', '-8.99'],
    ['3Y', '-18.78', '-57.64'],
    ['5Y', '-18.78', '-57.64'],
    ['Since inception', '-22.7', '-89.62'],
  ],
  [
    ['1Y', '17.48', '19.96'],
    ['3Y', '23.41', '24.25'],
    ['5Y', '19.9', '21.25'],
    ['Since inception', '20.83', '23.72'],
  ],
  [
    ['1Y', '0.13', '2.08'],
    ['3Y', '1.56', '-0.12'],
    ['5Y', '1.39', '0.11'],
    ['Since inception', '1.17', '-0.26'],
  ]
]

У меня получается только через циклы for в for и в for и дальше я просто уже теряюсь... А хотелось бы увидеть, как должно быть, чтоб изучить логику и понять как нужно.

Comments

[Aetae]

На lodash как-то так:

chain(t)
  .slice(1)
  .chunk(2)
  .map(chunk => zip(t[0], ...chunk))
  .value();

Но быстрее конечно будет на for'ах.)

Answer 1

[Rsa97]

const result = [];
for (let i = 1; i < t.length; i += 2) {
  const sub = [];
  for (j = 0; j < t[0].length; j += 1) {
    sub.push([t[0][j], t[i][j], t[i + 1][j]]);
  }
  result.push(sub);
}
console.log(result);

Comments

[Fedor Vlasenko]

Внутренний for здесь можно спокойно убрать. Длина массива известна. код станет компактнее

[Rsa97]

Fedor Vlasenko, Это если она известна и фиксирована. По входным данным больше похоже, что она может меняться.

Answer 2

[WbICHA]

Просто как вариант:

const columns = t[0];
  
  const result = t.reduce((acc, row1, i, arr) => {
    if(!(i & 1)) {
      return acc;
    }
    
    const row2 = arr[i + 1];
    const block = columns.map((c, j) => [c, row1[j], row2[j]]);
    
    acc.push(block);
    return acc;
  }, []);

Answer 3

[0xD34F]

const chunkedAndTransposed = ([ headers, ...arr ], chunkSize) =>
  arr.reduce((acc, n, i) => (
    (i % chunkSize) || acc.push(headers.map(m => [ m ])),
    n.forEach((m, j) => acc.at(-1)[j].push(m)),
    acc
  ), []);


const result = chunkedAndTransposed(arr, 2);

или, если на последний кусок данных не хватает, а размеры вложенных массивов должны быть одинаковыми, можно добавить null'ов или чего-нибудь ещё:

const chunkedAndTransposed = (arr, chunkSize, defautlValue = null) =>
  Array.from({ length: Math.ceil((arr.length - 1) / chunkSize) }, (_, iChunk) =>
    Array.from({ length: arr[0].length }, (_, iCol) =>
      Array.from({ length: chunkSize + 1 }, (_, iRow) =>
        iRow ? arr[iChunk * chunkSize + iRow]?.[iCol] ?? defautlValue : arr[0][iCol]
      )
    )
  );


const result = chunkedAndTransposed(arr, 5);

Comments

[WbICHA]

if (i--) {
    if (i % 2 === 0) {

Зачем изменять i? Можно же просто остаток от деления на 1 проверить.

[0xD34F]

Мне так больше нравится.

Шутка.

Когда писал, показалось, что уменьшенный индекс потребуется более одного раза. Не потребовался. Ну а -- остался.

UPD. В целом, какая-то позорная чушь получилась, не эстетично, так что полностью переписал.

[0xD34F]

Про "остаток от деления на 1" что-то не понял. Это как?

[WbICHA]

0xD34F, я имел ввиду i % 2 === 1.