从超类数组中筛选出子类

Question

我想要一个来自超类数组的子类数组。请参阅我的代码示例：

class A {}
class B: A {}

let arr: [A] = [A(), B()]

let arrOfBs: [B] = arr.filter { $0 as B } // Error here

如何正确过滤掉所有蜜蜂？

Answer 1

只需使用compactMap：

let arrOfBs = arr.compactMap { $0 as? B }

（或flatMap代表Swift＆lt; 4.1）

从技术上讲，你也可以使用：

let arrOfBs = arr.filter { $0 is B } as! [B]

但是compactMap更简单，没有令人讨厌的强制转换。

Answer 2

您正在过滤可以使用is而不是as

的数据类型

    class A {}
    class B: A {}

    let arr: [A] = [A(), B()]

    let arrOfBs = arr.filter { $0 is B } as? [B]
    print(arrOfBs?.count)

  

注意此结果可选，您可以使用if let或guard