Filter Array based on the Filtering of Another Array Swift

Question

Say I had two arrays:

var arrayA = ["Yes", "Yes2", "Not Answered", "No"]
var arrayB = ["Yes", "NA", "Yes2", "NA"]

And I wanted to remove the "NA"s from arrayB by doing:

var filtered = arrayB.filter({$0 != "NA"})

How could I remove the items at the same indexes removed in arrayA. I thought about using the find() function, but that only returns the first index a string appears. You can remove overlap from the arrays by:

let res = arrayA.filter { !contains(arrayB, $0) }

but how could I filter an array based on the filtering of another array?

the result would have:

arrayBFiltered = ["Yes", "Yes2"]
arrayAFiltered = ["Yes", "Not Answered"]

Any ideas?

You have accepted the answer I would consider the least "swift" way and the hackiest way. @Matteo's method is much nicer and much more "swifty". I do wonder why you don't have a data model to hold these data though? If they are so closely related then you should have them in a single model. — Fogmeister
– Fogmeister, Commented Jul 7, 2015 at 13:10

Matteo Piombo · Accepted Answer · 2015-07-07 14:40:56Z

6

You might prefer using zip:

var arrayA = ["Yes", "Yes2", "Not Answered", "No"]
var arrayB = ["Yes", "NA", "Yes2", "NA"]

let result = filter(zip(arrayA, arrayB)) { (a, b) in b != "NA" }

for (a, b) in result {
    println("A: \(a) -> B: \(b)")
}

EDIT: SWIFT 2.0

In Swift 2.0 it will be even easier to obtain an array of a consolidated ad-hod struct (say ... Foo) for clearer subsequent code:

struct Foo {
    let a: String
    let b: String
}
// Foo.init will be the function automatically generated by the default initialiser

let result = zip(arrayA, arrayB)
    .filter { (a, b) in b != "NA" }
    .map(Foo.init)
// Order of a and b is important

// result is an Array<Foo> suitable for a clearer subsequent code
for item in result {
    print("A: \(item.a) -> B: \(item.b)")
}

Hope this helps

edited Jul 7, 2015 at 14:40

answered Jul 7, 2015 at 5:28

Matteo Piombo

6,7362 gold badges27 silver badges25 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Fogmeister Over a year ago

This is what I was coming back to write. This is the "swift" way of doing this. Although, I question why the OP hasn't created an object to hold these related data?

Matteo Piombo Over a year ago

@Fogmeister As an exercice I added a Swift 2.0 variant ;-)

Community · Accepted Answer · 2017-05-23 12:16:48Z

1

The ideas from How can I sort multiple arrays based on the sorted order of another array could be used here, that would work for two or more arrays:

let arrayA = ["Yes", "Yes2", "Not Answered", "No"]
let arrayB = ["Yes", "NA", "Yes2", "NA"]

// Determine array indices that should be kept:
let indices = map(filter(enumerate(arrayB), { $1 != "NA" } ), { $0.0 } )

// Filter arrays based on the indices:
let arrayAFiltered = Array(PermutationGenerator(elements: arrayA, indices: indices))
let arrayBFiltered = Array(PermutationGenerator(elements: arrayB, indices: indices))

println(arrayAFiltered) // [Yes, Not Answered]
println(arrayBFiltered) // [Yes, Yes2]

edited May 23, 2017 at 12:16

CommunityBot

11 silver badge

answered Jul 7, 2015 at 5:39

Martin R

541k98 gold badges1.3k silver badges1.4k bronze badges

Comments

Qbyte · Accepted Answer · 2015-07-07 07:31:09Z

0

Another solution would be to make the remove code directly inside the filter closure:

// both are vars so you can mutate them directly
var arrayA = ["Yes", "Yes2", "Not Answered", "No"]
var arrayB = ["Yes", "NA", "Yes2", "NA"]

arrayA = filter(enumerate(arrayA)){
    arrayB.removeAtIndex($0)
    return $1 != "Na"
}

// use filtered arrayA and arrayB

answered Jul 7, 2015 at 7:31

Qbyte

13.3k5 gold badges46 silver badges60 bronze badges

Collectives™ on Stack Overflow

Filter Array based on the Filtering of Another Array Swift

3 Answers 3

2 Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

2 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related