Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Swift: Flatten an array of dictionaries to one dictionary

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

like image 911
Lneuner Avatar asked Feb 24 '16 09:02

Lneuner


2 Answers

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
    }
like image 108
Sergey Kalinichenko Avatar answered Sep 18 '22 19:09

Sergey Kalinichenko


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]
like image 31
Imanou Petit Avatar answered Sep 18 '22 19:09

Imanou Petit