LINQ按对象类型选择

Question

我有一个包含两种对象A和B的集合.

Class Base{}
Class A : Base {}
Class B : Base {}

List<Base> collection = new List<Base>();
collection.Add(new A());
collection.Add(new B());
collection.Add(new A());
collection.Add(new A());
collection.Add(new B());

现在我想根据集合的类型(A或B,而不是两者)选择集合中的对象.

我怎么能为此编写LINQ查询？请帮我.否则我需要遍历集合,这是我不想要的.谢谢.

编辑:

感谢大家的帮助.现在我可以使用LINQ中的OfType().但我认为在我的情况下它不会工作.我的情况是

Class Container
{
  List<Base> bases;
}

List<Container> containers = new List<Container>();

现在我想从容器中选择容器,容器至少有一个类型A.可能这不能由LINQ完成.非常感谢.

Answer 1

您可以使用OfTypeLinq方法:

var ofTypeA = collection.OfType<A>();

关于你不愿意循环收集,你应该记住,Linq不做魔术; 我没有检查执行情况OfType,但我会感到很惊讶没有找到一个循环或迭代器在那里.

Answer 2

您可以使用OfType扩展方法

IEnumerable<A> filteredToA = list.OfType<A>();
IEnumerable<B> filteredToB = list.OfType<B>();

Answer 3

为了完整性,这里是源代码Enumerable.OfType<T>.

public static IEnumerable<TResult> OfType<TResult>(this IEnumerable source) {
    if (source == null) throw Error.ArgumentNull("source"); 
    return OfTypeIterator<TResult>(source); 
}

static IEnumerable<TResult> OfTypeIterator<TResult>(IEnumerable source) {
    foreach (object obj in source) {
        if (obj is TResult) yield return (TResult)obj;
    } 
}

您可以看到它懒惰地评估源流.