แมปอาร์เรย์ของวัตถุกับพจนานุกรมใน Swift

ฉันมีอาร์เรย์ของPersonวัตถุ:

class Person {
   let name:String
   let position:Int
}

และอาร์เรย์คือ:

let myArray = [p1,p1,p3]

ฉันต้องการแมปmyArrayให้เป็นพจนานุกรมของ[position:name]โซลูชันแบบคลาสสิกคือ:

var myDictionary =  [Int:String]()

for person in myArray {
    myDictionary[person.position] = person.name
}

มีวิธีใดที่สวยงามโดย Swift ในการทำเช่นนั้นด้วยแนวทางการใช้mapงานflatMap... หรือสไตล์ Swift สมัยใหม่อื่น ๆ

โอเคmapไม่ใช่ตัวอย่างที่ดีเพราะมันเหมือนกับการวนซ้ำคุณสามารถใช้reduceแทนได้โดยเอาวัตถุแต่ละชิ้นมารวมกันและเปลี่ยนเป็นค่าเดียว:

let myDictionary = myArray.reduce([Int: String]()) { (dict, person) -> [Int: String] in
    var dict = dict
    dict[person.position] = person.name
    return dict
}

//[2: "b", 3: "c", 1: "a"]

ใน Swift 4 หรือสูงกว่าโปรดใช้คำตอบด้านล่างเพื่อให้ได้ไวยากรณ์ที่ชัดเจนยิ่งขึ้น

เนื่องจากSwift 4คุณสามารถทำแนวทางของ @ Tj3n ได้อย่างหมดจดและมีประสิทธิภาพมากขึ้นโดยใช้intoเวอร์ชันของreduceมันจะกำจัดพจนานุกรมชั่วคราวและค่าที่ส่งคืนเพื่อให้อ่านได้เร็วขึ้นและง่ายขึ้น

การตั้งค่าโค้ดตัวอย่าง:

struct Person { 
    let name: String
    let position: Int
}
let myArray = [Person(name:"h", position: 0), Person(name:"b", position:4), Person(name:"c", position:2)]

Into พารามิเตอร์ถูกส่งผ่านพจนานุกรมว่างเปล่าของประเภทผลลัพธ์:

let myDict = myArray.reduce(into: [Int: String]()) {
    $0[$1.position] = $1.name
}

ส่งคืนพจนานุกรมประเภทที่ส่งเข้ามาโดยตรงinto:

print(myDict) // [2: "c", 0: "h", 4: "b"]

ตั้งแต่Swift 4คุณสามารถทำได้อย่างง่ายดาย มีตัวเริ่มต้นใหม่สอง ตัวที่สร้างพจนานุกรมจากลำดับของทูเปิล (คู่ของคีย์และค่า) หากรับประกันว่าคีย์ไม่ซ้ำกันคุณสามารถดำเนินการดังต่อไปนี้:

let persons = [Person(name: "Franz", position: 1),
               Person(name: "Heinz", position: 2),
               Person(name: "Hans", position: 3)]

Dictionary(uniqueKeysWithValues: persons.map { ($0.position, $0.name) })

=> [1: "Franz", 2: "Heinz", 3: "Hans"]

สิ่งนี้จะล้มเหลวด้วยข้อผิดพลาดรันไทม์หากคีย์ใด ๆ ซ้ำกัน ในกรณีนี้คุณสามารถใช้เวอร์ชันนี้:

let persons = [Person(name: "Franz", position: 1),
               Person(name: "Heinz", position: 2),
               Person(name: "Hans", position: 1)]

Dictionary(persons.map { ($0.position, $0.name) }) { _, last in last }

=> [1: "Hans", 2: "Heinz"]

สิ่งนี้จะทำงานเป็นห่วงของคุณ หากคุณไม่ต้องการ "เขียนทับ" ค่าและยึดติดกับการแมปแรกคุณสามารถใช้สิ่งนี้:

Dictionary(persons.map { ($0.position, $0.name) }) { first, _ in first }

=> [1: "Franz", 2: "Heinz"]

Swift 4.2เพิ่มinitializer ตัวที่สามที่จัดกลุ่มองค์ประกอบตามลำดับลงในพจนานุกรม คีย์พจนานุกรมได้มาจากการปิด องค์ประกอบที่มีคีย์เดียวกันจะถูกใส่ลงในอาร์เรย์ในลำดับเดียวกันกับในลำดับ สิ่งนี้ช่วยให้คุณได้ผลลัพธ์ที่คล้ายกันดังข้างต้น ตัวอย่างเช่น:

Dictionary(grouping: persons, by: { $0.position }).mapValues { $0.last! }

=> [1: Person(name: "Hans", position: 1), 2: Person(name: "Heinz", position: 2)]

Dictionary(grouping: persons, by: { $0.position }).mapValues { $0.first! }

=> [1: Person(name: "Franz", position: 1), 2: Person(name: "Heinz", position: 2)]

คุณสามารถเขียนโปรแกรมเริ่มต้นที่กำหนดเองสำหรับDictionaryประเภทตัวอย่างเช่นจากสิ่งต่อไปนี้:

extension Dictionary {
    public init(keyValuePairs: [(Key, Value)]) {
        self.init()
        for pair in keyValuePairs {
            self[pair.0] = pair.1
        }
    }
}

จากนั้นใช้mapสำหรับอาร์เรย์ของคุณPerson:

var myDictionary = Dictionary(keyValuePairs: myArray.map{($0.position, $0.name)})

แล้วโซลูชันที่ใช้ KeyPath ล่ะ?

extension Array {

    func dictionary<Key, Value>(withKey key: KeyPath<Element, Key>, value: KeyPath<Element, Value>) -> [Key: Value] {
        return reduce(into: [:]) { dictionary, element in
            let key = element[keyPath: key]
            let value = element[keyPath: value]
            dictionary[key] = value
        }
    }
}

นี่คือวิธีที่คุณจะใช้:

struct HTTPHeader {

    let field: String, value: String
}

let headers = [
    HTTPHeader(field: "Accept", value: "application/json"),
    HTTPHeader(field: "User-Agent", value: "Safari"),
]

let allHTTPHeaderFields = headers.dictionary(withKey: \.field, value: \.value)

// allHTTPHeaderFields == ["Accept": "application/json", "User-Agent": "Safari"]

คุณสามารถใช้ฟังก์ชันลด ก่อนอื่นฉันได้สร้าง initializer ที่กำหนดไว้สำหรับคลาส Person

class Person {
  var name:String
  var position:Int

  init(_ n: String,_ p: Int) {
    name = n
    position = p
  }
}

ต่อมาฉันได้เริ่มต้นอาร์เรย์ของค่า

let myArray = [Person("Bill",1), 
               Person("Steve", 2), 
               Person("Woz", 3)]

และสุดท้ายตัวแปรพจนานุกรมมีผลลัพธ์:

let dictionary = myArray.reduce([Int: Person]()){
  (total, person) in
  var totalMutable = total
  totalMutable.updateValue(person, forKey: total.count)
  return totalMutable
}

นี่คือสิ่งที่ฉันได้ใช้

struct Person {
    let name:String
    let position:Int
}
let persons = [Person(name: "Franz", position: 1),
               Person(name: "Heinz", position: 2),
               Person(name: "Hans", position: 3)]

var peopleByPosition = [Int: Person]()
persons.forEach{peopleByPosition[$0.position] = $0}

คงจะดีถ้ามีวิธีรวม 2 บรรทัดสุดท้ายเพื่อให้peopleByPositionเป็นไฟล์let.

เราสามารถขยาย Array ที่ทำเช่นนั้นได้!

extension Array {
    func mapToDict<T>(by block: (Element) -> T ) -> [T: Element] where T: Hashable {
        var map = [T: Element]()
        self.forEach{ map[block($0)] = $0 }
        return map
    }
}

จากนั้นเราก็ทำได้

let peopleByPosition = persons.mapToDict(by: {$0.position})

extension Array {
    func mapToDict<T>(by block: (Element) -> T ) -> [T: Element] where T: Hashable {
        var map = [T: Element]()
        self.forEach{ map[block($0)] = $0 }
        return map
    }
}

อาจจะเป็นแบบนี้?

myArray.forEach({ myDictionary[$0.position] = $0.name })