Ruby技术问题：如何将数组指定元素重组为新数组？

嘿，我猜你大概率是在遍历数组的时候搞反了循环顺序，或是没正确初始化分组数组！先别急，咱们先唠唠这类需求里最容易踩的坑，再给你靠谱的解决方案～

• 循环顺序搞反：比如你可能先挨个遍历每个输入数组的元素，再处理位置，这样会把所有数组的元素按顺序堆在一起，而不是按对应位置分组。举个典型的错误示例：

// 错误示范：循环顺序颠倒
const input = [[1,2,3], [4,5,6], [7,8,9]];
const result = [];
for (let arr of input) {
  for (let i = 0; i < arr.length; i++) {
    result.push(arr[i]);
  }
}
// 得到的结果是 [1,2,3,4,5,6,7,8,9]，完全不符合按位置分组的需求

• 未初始化分组数组：当你按位置遍历的时候，没提前给result[i]创建空数组就直接push元素，要么会报错，要么会把元素塞到错误的位置，导致结果混乱。
• 忽略不等长数组边界：如果输入的数组长度不一样，没判断当前位置是否存在元素，会把undefined混进结果里，造成数据混乱。

假设我们用JavaScript来实现，这里有两种简洁靠谱的写法：

方法1：按最大长度遍历位置（直观易懂） #

const input = [[1,2,3], [4,5,6], [7,8,9]];
const result = [];
// 先找到最长的输入数组长度，确保所有位置都能覆盖到
const maxLength = Math.max(...input.map(arr => arr.length));

for (let i = 0; i < maxLength; i++) {
  // 为每个位置初始化一个空数组
  result[i] = [];
  for (let arr of input) {
    // 只添加存在的元素，避免不等长数组混入undefined
    if (arr[i] !== undefined) {
      result[i].push(arr[i]);
    }
  }
}

console.log(result);
// 输出：[[1,4,7], [2,5,8], [3,6,9]]，完全符合需求！

方法2：用reduce简化代码（函数式风格） #

如果你偏爱函数式写法，可以用Array.prototype.reduce来简化逻辑：

const input = [[1,2,3], [4,5,6], [7,8,9]];
const result = input.reduce((acc, currentArr) => {
  currentArr.forEach((item, index) => {
    // 若当前位置的分组数组不存在，先初始化空数组
    if (!acc[index]) acc[index] = [];
    acc[index].push(item);
  });
  return acc;
}, []);

console.log(result);
// 同样得到目标结果：[[1,4,7], [2,5,8], [3,6,9]]

要是你能把自己当前的代码贴出来，我还能帮你精准定位具体哪里出问题哦！

内容的提问来源于stack exchange，提问作者A Cox