Swift - 数组到字典

Question

所以，我有以下对象：

struct Person {
   let name: String
   let birthday: String
}

let Bob = Person(name: "Bob", birthday: "11.12.1987")
let Tim = Person(name: "Tim", birthday: "11.12.1987")
let John = Person(name: "John", birthday: "01.02.1957")
let Jerry = Person(name: "Jerry", birthday: "17.12.2001")

还有以下数组：

let people = [Bob, Tim, John, Jerry]

我的目标是从这个数组中生成一个字典，其中键为“生日”，值为“Person”对象本身：[String: [Person]]。如果有相同的键，则应添加该人并形成一个数组作为键。所以结果如下：

dictionary = ["11.12.1987": [Bob, Tim], "01.02.1957": John, "17.12.2001": Jerry]

实现这一目标的最佳方法是什么？

干杯！

Answer 1

您可以将Dictionary(grouping:by:) 方法与mapValues 一起使用，

let result = Dictionary(grouping: people, by: \.birthday)

print(result)

输出是，

[“01.02.1957”：[“约翰”]，“17.12.2001”：[“杰瑞”]，“11.12.1987”： [“鲍勃”，“蒂姆”]]

这将为您生成Dictionary<String, [Person]> 类型的字典。

从你的问题来看，如果有多个人时只有一个人和数组，你似乎想要一个人。在那种情况下，你会丢失类型，你的类型会是这样的，Dictionary<String, Any>。你可以对上面的代码做一点修改来做到这一点。

let result = Dictionary(grouping: people, by: \.birthday)
    .mapValues { people -> Any in
                if people.count == 1 {
                    return people.first!
                }
                return people
            }

print(result)

输出将与您想要的相似，

[“01.02.1957”：“约翰”，“17.12.2001”：“杰瑞”，“11.12.1987”：[“鲍勃”， “蒂姆”]]

Answer 2

如何通过字典循环检查日期是否存在。否则创建带有日期的键并赋值。 如果 key 已经存在，则将 person 添加到实际的 key 值：

var dictionary = [String: [Person]]()
        
        // Loop through all person objects in array
        for person in people {
            
            // Check if date already exists
            if dictionary[person.birthday] != nil {
                // If already exists, include new person in array
                var array = dictionary[person.birthday]
                array!.append(person)
            }
            // If date does not exists, add new key with date and person
            else {
                dictionary[person.birthday] = [person]
            }
            
        }