JavaScript 中 Ruby 的 each_cons 等价物

Question

该问题已针对多种语言提出，但未针对 javascript。

Ruby 有 Enumerable#each_cons 方法，看起来像这样：

puts (0..5).each_cons(2).to_a
# [[0, 1], [1, 2], [2, 3], [3, 4], [4, 5]]
puts (0..5).each_cons(3).to_a
# [[1, 2, 3], [2, 3, 4], [3, 4, 5]]

我怎么能在 Array 的 javascript 中使用类似的方法？

Answer 1

这是一个可以做到的函数（ES6+）：

// functional approach
const eachCons = (array, num) => {
    return Array.from({ length: array.length - num + 1 },
                      (_, i) => array.slice(i, i + num))
}

// prototype overriding approach
Array.prototype.eachCons = function(num) {
  return Array.from({ length: this.length - num + 1 },
                    (_, i) => this.slice(i, i + num))
}


const array = [0,1,2,3,4,5]
const log = data => console.log(JSON.stringify(data))

log(eachCons(array, 2))
log(eachCons(array, 3))

log(array.eachCons(2))
log(array.eachCons(3))

你必须猜测结果数组的长度 (n = length - num + 1)，然后你可以利用 JavaScript 的 array.slice 来获得你需要的块，迭代 n 次。

Answer 2

您可以获取长度，构建一个新数组并使用Array.from 和内置映射器映射切片数组。

Number.prototype[Symbol.iterator] = function* () {
    for (var i = 0; i < this; i++) yield i;
};

Array.prototype.eachCons = function (n) {
    return Array.from({ length: this.length - n + 1}, (_, i) => this.slice(i, i + n));
}

console.log([...10].eachCons(2));
console.log([...10].eachCons(3));

.as-console-wrapper { max-height: 100% !important; top: 0; }

Answer 3

这是一个依赖函数生成器的单行解决方案。可能略有不同，但仍然值得一试。

我也添加了原型，不知道你为什么要这样做，但仍然..

这个解决方案不需要 ES6，它也可以和 ES5 一起使用，稍微注意一点不支持函数生成器的 IE。

function* eachCons(arr, num) {
  for (let counter = 0; counter <= arr.length - num; counter++) yield arr.slice(counter, counter + num);
}

Array.prototype.eachCons = function* (num) {
  for (let counter = 0; counter <= this.length - num; counter++) yield this.slice(counter, counter + num);
}

const array = [0,1,2,3,4,5];
const log = data => console.log(JSON.stringify(data))
log([...eachCons(array, 2)]);
log([...eachCons(array, 3)]);

// Iterable way
for (let [...pack] of eachCons(array, 2)) {
  console.log('pack is', pack);
}
// Prototype...
for (let [...pack] of array.eachCons(2)) {
  console.log('pack is', pack);
}