c#Enumerable采用谓词和结果限制的方法

问题描述 投票:-2回答:2

在C#for IEnumerable中是否有任何扩展方法可以使用谓词返回集合中的项目,直到达到某个给定的限制为止?

例如:

string[] source = { "zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine" };
var result = source.Take(2, item => item.EndsWith("e"));
// result == { "one", "three" }

这个例子很容易反映这个想法,但是对于更复杂的逻辑,它可能是有用的,而不是评估集合中的每个项目。

我提出了以下解决方案,但也许.NET中已经有类似的东西了。

请勿使用以下代码 - 仅供参考

public static IEnumerable<T> Take<T>(this IEnumerable<T> enumerable,
    int count,
    Func<T, bool> predicate)
{
    if (enumerable == null)
    {
        yield break;
    }

    using (IEnumerator<T> iterator = enumerable.GetEnumerator())
    {
        var matchingCount = 0;
        while (matchingCount < count && iterator.MoveNext())
        {
            if (predicate(iterator.Current))
            {
                matchingCount++;
                yield return iterator.Current;
            }
        }
    }
}

UPDATE

优雅的解决方案 - 这也会做同样的事情:

public static IEnumerable<T> Take<T>(this IEnumerable<T> enumerable,
    int count,
    Func<T, bool> predicate)
{
    return enumerable.Where(predicate).Take(count);
}

所以通过执行以下代码:

string[] source = { "zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine" }; 
var result = source
    .Where(n => { 
        Console.WriteLine("processing " + n); 
        return n.EndsWith("e"); 
    })
   .Take(2)
   .ToList();

以下内容将记录在控制台中:

processing one
processing three

希望这将有助于未来的任何人。

记住人:不要重新发明轮子。

c# linq ienumerable
2个回答
6
投票

你可以结合TakeWhere

var result = source.Where(item => item.Contains("e")).Take(2);

Where将过滤集合,Take将在找到2个元素后停止迭代。因为WhereTake都是懒惰的,所以在迭代之前不会创建任何集合,就像你的解决方案一样,可能存在性能损失,因为我们有两个LINQ运算符,但在大多数情况下这应该不是问题。


0
投票

有:

var results = source.Where(predicate).Take(limit);
© www.soinside.com 2019 - 2024. All rights reserved.