1. 首页
  2. 问题
  3. 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)吗？

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


阅读 257

收藏
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