对于IEnumerable,C#中是否有任何扩展方法可以使用谓词返回集合中的项目,直到达到某个给定的限制为止?
例如:
IGetsSomeData
这个例子很容易反映这个想法,但是对于更复杂的逻辑,它可能是有用的,而不是评估集合中的每个项目。
我提出了以下解决方案,但也许.NET中已经存在类似的东西了。
请勿使用以下代码 - 仅供参考
string[] source = { "zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine" };
var result = source.Take(2, item => item.EndsWith("e"));
// result == { "one", "three" }
更新
优雅的解决方案 - 这样做也是如此:
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;
}
}
}
}
通过执行以下代码:
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();
希望这将有助于将来的任何人。
请记住:不要重新发明轮子。
答案 0 :(得分:6)
您可以合并Take
和Where
:
var result = source.Where(item => item.Contains("e")).Take(2);
Where
while会过滤集合,Take
会在找到2个元素后停止迭代。由于Where
和Take
都是惰性的,因此在迭代之前不会创建任何集合,就像您的解决方案一样,可能存在性能损失,因为我们有两个LINQ运算符,但在大多数情况下,这不应该是一个问题。
答案 1 :(得分:0)
有:
var results = source.Where(predicate).Take(limit);