迭代并返回连续 n 个元素的数组

Question

在 Ruby 中，Enumerable 上有一个 each_cons一个>。它的工作原理是这样的

(1..5).each_cons(3) {|n| p n}

[1,2,3]
[2,3,4]
[3,4,5]

我想用 C# 来做这件事。 LINQ 会很好。

下面的代码做了类似的事情，但它循环一对多，并且还被硬编码为仅返回两个元素

var ints = new int[] { 1, 2, 3, 4, 5, 6, 7 };
var cons = ints.Select((o, i) =>
            new int[]{ ints[i], i == ints.Length - 1 ? 0 : ints[i + 1] });

。如果可以将其创建为原始数组的迭代器，而不必创建大量数组，那就太好了。

Answer 1

尝试以下操作

var ints = Enumerable.Range(1, 3).Select(x => Enumerable.Range(x, 3));

这将返回具有指定值的 IEnumerable> 。如果这是意图的话，您可以随时添加 .ToArray 表达式以将其放入数组中（无法判断这是否是 ruby​​ 中 [] 的含义）

Answer 2

您可以创建一个扩展方法，以这种方式实现它：

    public static IEnumerable<IEnumerable<T>> each_cons<T>(this IEnumerable<T> enumerable, int length)
    {
        for (int i = 0; i < enumerable.Count() - length + 1; i++)
        {
            yield return enumerable.Skip(i).Take(length);
        }
    }

使用它：

var ints = Enumerable.Range(1, 7).each_cons(3);

Answer 3

这是一个通用的扩展方法，对于我当前的用例来说，它有点复杂，但它似乎有效。

static class EnumerableExtension
{
    public static IEnumerable<IEnumerable<T>> EachCons<T>(this IEnumerable<T> sequence, int cons)
    {
        for (IEnumerable<T> head = sequence.Take(cons), tail = sequence.Skip(1);
             head.Count() == cons; head = tail.Take(cons), tail = tail.Skip(1))
             yield return head;
    }
}

改用 Jay 的实现。速度快得多。我刚刚把这个留在这里，因为杰伊的回答中提到了它。