Swift: Flatten an array of dictionaries to one dictionary
Asked Answered
W

4

18

In Swift, I am trying to flatten an array of dictionaries into one dictionary i.e

let arrayOfDictionaries = [["key1": "value1"], ["key2": "value2"], ["key3": "value3", "key4": "value4"]]


//the end result will be:   
 flattenedArray = ["key1": "value1", "key2": "value2", "key3": "value3", "key4": "value4"]

I have tried using flatmap, but the type of the returned result is [(String, AnyObject)] and not [String, Object] ie

let flattenedArray = arrayOfDictionaries.flatMap { $0 }
// type is [(String, AnyObject)]

So I have 2 questions:

  • Why is type [(String, AnyObject)] returned? And what do the brackets mean?

  • How do I achieve the desired result?

Edit: I would prefer to use a functional approach with Swift's map/flatmap/reduce etc. instead of a for-loop

Willawillabella answered 24/2, 2016 at 9:15 Comment(0)
A
19

what do the brackets mean?

This, along with a comma instead of a colon, should provide the first clue: brackets mean that you get an array of tuples. Since you are looking for a dictionary, not an array, this tells you that you need to convert the sequence of tuples (key-value pairs) to a single dictionary.

How do I achieve the desired result?

One way to do it would be using reduce, like this:

let flattenedDictionary = arrayOfDictionaries
    .flatMap { $0 }
    .reduce([String:String]()) { (var dict, tuple) in
        dict.updateValue(tuple.1, forKey: tuple.0)
        return dict
    }
Antonia answered 24/2, 2016 at 9:45 Comment(10)
Sir you did little mistake i have fixed it :) Please check the EditDedrick
Here there was problem "dict.updateValue(tuple.0, forKey: tuple.1)" i changed to dict.updateValue(tuple.1, forKey: tuple.0)Dedrick
Thanks for this. I feel like there could be a slightly more elegant way though. Do you know why flatmap is mapping it to an array of tuples and not an array of dictionaries?Willawillabella
@Willawillabella Yes, (String, AnyObject) in Swift's type notation denotes a tuple. A map would use brackets and a colon, i.e. [String:AnyObject].Antonia
@dasblinkenlight but why is it mapping it to a tuple instead of dictionary?Willawillabella
@Willawillabella Because map/flatMap iterate the dictionary. When a dictionary gets iterated, each item in the dictionary is returned as a (key, value) tuple. This is described here at the bottom of the document.Antonia
@dasblinkenlight thank you! That answers my questionWillawillabella
@dasblinkenlight one more thing. I am slightly confused as to what [String:String]()) { (var dict, tuple) actually does in the closure. Could you point me to a good resource where I could read more about it? :)Willawillabella
@Willawillabella reduce takes two things - the "initial value" of the result, and the function that "adds" the current element to the result. In this case reduce starts with an empty dictionary, hence the [String:String]() constructor. (var dict, tuple) is the header of the lambda function that adds the current tuple to the dictionary.Antonia
@dasblinkenlight great explanation! Thanks!Willawillabella
H
17

With Swift 5, Dictionay has a init(_:uniquingKeysWith:) initializer. init(_:uniquingKeysWith:) has the following declaration:

init<S>(_ keysAndValues: S, uniquingKeysWith combine: (Value, Value) throws -> Value) rethrows where S : Sequence, S.Element == (Key, Value)

Creates a new dictionary from the key-value pairs in the given sequence, using a combining closure to determine the value for any duplicate keys.


The two following Playground sample codes show how to flatten an array of dictionaries into a new dictionary.

let dictionaryArray = [["key1": "value1"], ["key1": "value5", "key2": "value2"], ["key3": "value3"]]

let tupleArray: [(String, String)] = dictionaryArray.flatMap { $0 }
let dictonary = Dictionary(tupleArray, uniquingKeysWith: { (first, last) in last })

print(dictonary) // prints ["key2": "value2", "key3": "value3", "key1": "value5"]
let dictionaryArray = [["key1": 10], ["key1": 10, "key2": 2], ["key3": 3]]

let tupleArray: [(String, Int)] = dictionaryArray.flatMap { $0 }
let dictonary = Dictionary(tupleArray, uniquingKeysWith: { (first, last) in first + last })
//let dictonary = Dictionary(tupleArray, uniquingKeysWith: +) // also works

print(dictonary) // ["key2": 2, "key3": 3, "key1": 20]
Hangover answered 13/6, 2017 at 22:34 Comment(1)
Why were these Swift 4 features built for "initialization from a sequence of (Key, Value) tuples" rather (or also) than "a sequence of dictionaries"? That forces inclusion of the tupleArray step in your code, which is kind of annoying.Seto
J
9

Updating @dasblinkenlight's answer for Swift 3.

"var" in the parameters has been deprecated, but this approach worked fine for me.

let flattenedDictionary = arrayOfDictionaries
    .flatMap { $0 }
    .reduce([String:String]()) { (dict, tuple) in
        var nextDict = dict
        nextDict.updateValue(tuple.1, forKey: tuple.0)
        return nextDict
    }
Jilt answered 21/4, 2017 at 6:42 Comment(1)
the nextDict should be a var not letBeneficence
D
1

Here is the way to do

let arrayOfDictionaries = [["key1": "value1"], ["key2": "value2"], ["key3": "value3", "key4": "value4"]]
var dic = [String: String]()
for item in arrayOfDictionaries {
    for (kind, value) in item {
        print(kind)
        dic.updateValue(value, forKey: kind)
    }


}
print(dic)

print(dic["key1"]!)

OUTPUT

OUTPUT

Dedrick answered 24/2, 2016 at 9:44 Comment(1)
Thanks for this. While this approach is correct, I would prefer to use a functional approach with Swift's map/flatmap/reduce etc.Willawillabella

© 2022 - 2024 — McMap. All rights reserved.