Swift-如何获取数组过滤项的索引

小编典典

Swift-如何获取数组过滤项的索引

swift

let items: [String] = ["A", "B", "A", "C", "A", "D"]

items.whatFunction("A") // -> [0, 2, 4]
items.whatFunction("B") // -> [1]

Swift 3支持像这样的功能whatFunction(_: Element)吗？

如果没有，最有效的逻辑是什么？

阅读 349

2020-07-07

共1个答案

小编典典

您可以为数组创建自己的扩展名。

extension Array where Element: Equatable {
    func indexes(of element: Element) -> [Int] {
        return self.enumerated().filter({ element == $0.element }).map({ $0.offset })
    }
}

您可以这样称呼它

items.indexes(of: "A") // [0, 2, 4]
items.indexes(of: "B") // [1]

2020-07-07