13

I just discovered the Swift zip function recently. It seems quite useful.

It takes 2 input arrays and creates an array of tuples out of pairs of values from each array.

Is there a variant of zip that takes an arbitrary number of arrays and outputs tuples with that same number of elements? It seems like there should be a way to do this.

Duncan C
  • 122,346
  • 22
  • 162
  • 258
  • I'm no expert at Swift but while you can create a function that takes a variable number of sequence parameters, how would you declare the return type to be a tuple of unknown size? – rmaddy Nov 09 '16 at 23:45
  • Good point. You can use generics to define the function to create tuples who's elements are the types of the input array elements, but I don't know of a way in Swift to define a tuple of variable size. – Duncan C Nov 09 '16 at 23:49

3 Answers3

19

Bear in mind, you can nest one zip inside another, and then unpack it with a nested tuple:

let integers = [1, 2, 3, 4, 5]
let strings = ["a", "b", "c", "d", "e"]
let doubles = [1.0, 2.0, 3.0, 4.0, 5.0]

for (integer, (string, double)) in zip(integers, zip(strings, doubles)) {
    print("\(integer) \(string) \(double)")
}

Not quite as elegant as having a zip for arbitrary n-tuples, but it gets the job done.

Airspeed Velocity
  • 39,686
  • 7
  • 107
  • 114
11

No, zip for an arbitrary number of sequences isn't currently possible due to Swift's lack of variadic generics. This is discussed in the Generics Manifesto.

In the meanwhile, I wrote a gyb template for generating ZipSequences of custom arity. I've also pre-generated ZipSequences of arity 3...10 for your convenience. It's available here.

In action:

let integers = [1, 2, 3, 4, 5]
let strings = ["a", "b", "c", "d", "e"]
let doubles = [1.0, 2.0, 3.0, 4.0, 5.0]

for (integer, string, double) in zip(integers, strings, doubles) {
    print("\(integer) \(string) \(double)")
}

Prints:

1 a 1.0

2 b 2.0

3 c 3.0

4 d 4.0

5 e 5.0

Community
  • 1
  • 1
Alexander
  • 54,014
  • 10
  • 87
  • 136
0

You can use the nested zip approach as in some of the previous answers, and use map to flatten the result:

let combined = zip(integers, zip(strings, doubles)).map { ( $0.0, $0.1.0, $0.1.1 ) }

It yields an array of tuples, each with three elements. Looks a bit ugly but as long as it works… It does work as long as the number of arrays to zip is fixed.

Tom E
  • 1,408
  • 7
  • 15