swift 配列 削除

Question

配列について質問です。

やりたい内容としては、

var nums: [Int] = [2,3,4,1,5]

4という要素を探して、4という要素があれば削除。
みたいにしたいのですが出来るのでしょうか？調べてもindexでの削除しか見当たりません。

Comments

[fuzzball]

4が複数あった場合はどうするのでしょうか？

[tosi_1211]

複数削除したいです！

Answer 1

以下のようなextensionを定義すると便利かと思います。
参考URL : 使うと手放せなくなるSwift Extension集

var nums: [Int] = [2,3,4,1,5]

extension Array where Element: Equatable {
    
    mutating func remove(element element: Element) -> Bool {
        guard let index = indexOf(element) else { return false }
        removeAtIndex(index)
        return true
    }
    
    mutating func e_remove(elements elements: [Element]) {
        for element in elements {
            remove(element: element)
        }
    }
}

nums.remove(element: 4)

nums
//=> [2, 3, 1, 5]

Comments

[tosi_1211]

ありがとうございます！Extensionとても勉強になります！

Answer 2

nums = nums.filter {$0 != 4}