Swift Dictionary型のvalueに配列を作った場合にappendする方法

Question

swift で Dictionary型のvalueに配列をとる場合に、
各keyのvalueに対してappendする方法がわかりません。

下記のような場合を想定しています。

あるObjectクラスが存在します。
そのObjectのインスタンスを持つ配列objectArrayがあります。

DictionaryのkeyにObject.nameを設定します。
valueにはobjectインスタンスをappendします。

class Object {
   var Name: String = ""
   var Value:Int = 0
   init(name:String, value:Int){
       Name = name
       Value = value
   }
}

var dic:[String:[Object]] = [:]

let obj1 = Object(name: "name1", value: 1)
let obj2 = Object(name: "name2", value: 2)
let obj3 = Object(name: "name3", value: 3)
let obj4 = Object(name: "name2", value: 4)
let obj5 = Object(name: "name4", value: 5)
let obj6 = Object(name: "name3", value: 6)

let objectArray:[Object] = [ obj1, obj2, obj3, obj4, obj5, obj6]

for object in objectArray{
    dic[object.name]?.append(object)
}

最終的に
dic["test"][object1, object2, object3]
dic["test2"][object4]
というような結果が欲しいです。

しかし、上記のようなコードを実行するとdicに値が入ってきません。nilになります。
どうしたら想定の結果が得られるでしょうか。
よろしくお願いします！

Answer 1

配列の初期化をしてからでないとappendされないのでnilになります。

class Object {
    var name: String = ""
    var value: Int = 0
}

var dic: [String: [Object]] = [:]

var obj1 = Object()
obj1.name = "test"
obj1.value = 1

var obj2 = Object()
obj2.name = "test"
obj2.value = 2

var obj3 = Object()
obj3.name = "test"
obj3.value = 3

var objectArray: [Object] = [obj1, obj2, obj3]

for object in objectArray {
    
    // nilの場合配列を初期化する
    dic[object.name] = dic[object.name] ?? []
    dic[object.name]?.append(object)
}

print(dic["test"])
//=> Optional([Object, Object, Object])

print(dic["test"]![2].value)
//=> 3

Comments

[yamahare]

おおおおー！！すごく綺麗なコードだあー
私なりに考えたコードは、ifとelse使ってて微妙でしたが、こっちは格好良いです！
とても参考になりました。ありがとうございます！！！＼(^o^)／

Answer 2

dicに値が入らない理由は、
dic[object.name]?.append(object)を実行する時のdic[object.name]がnilだからです。
nilにappendしようとしても無視されます。
この行を実行する前に、dic[object.name]がnilだったら、その値を配列エリアとして初期化する処理が必要だと思います。
以下のようにすれば、おそらくあなたのやりたいことができると思います。

for object in objectArray {
    if (dic[object.name] == nil) {
        dic[object.name] = []
    }
    dic[object.name]?.append(object)
}

Comments

[yamahare]

ありがとうございます。TakeOneさんの説明を受けてより理解がすすみました。やはりnilになっていたのですね！（・ω・）参考にさせていただきます！

[TakeOne]

すいません。回答を書いている間に解決していましたね(^_^;)

Answer 3

とりあえず下記のように、dic[object.name]がnilのときは[object]をいれてあげると、動作しました！

for object in objectArray{
   if dic[object.name] == nil {
       dic[object.name] = [object]
   }else{
       dic[object.name]?.append(object)
   }
}