Map Swift array with specific format

Question

Given an array that contains elements like this:

let array = [[a], [b, c], [d, e, f]]

Is there an elegant way to convert this array to an array which returns a tuple with the index of the outer array:

let result = [(a, 0), (b, 1), (c, 1), (d, 2), (e, 2), (f, 2)]

@holex I can do this with for loop，but this could be more elegant. — Zigii Wong
– Zigii Wong, Commented Dec 19, 2017 at 15:21

Ole Begemann · Accepted Answer · 2017-12-19 14:03:59Z

9

let array = [["a"], ["b", "c"], ["d", "e", "f"]]
let result = zip(array, array.indices).flatMap { subarray, index in
    subarray.map { ($0, index) }
}

result is:

[("a", 0), ("b", 1), ("c", 1), ("d", 2), ("e", 2), ("f", 2)]

I used zip(array, array.indices) instead of array.enumerated() because you specifically asked for a tuple with the array index — enumerated() produces tuples with zero-based integer offsets. If your source collection is an array, it doesn't make a difference, but other collections (like ArraySlice) would behave differently.

edited Dec 19, 2017 at 14:03

answered Dec 19, 2017 at 13:55

Ole Begemann

136k32 gold badges281 silver badges258 bronze badges

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

Collectives™ on Stack Overflow

Map Swift array with specific format

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related