Python 辞書 append str

Question

Pythonで名前と年齢を管理するプログラムを書いています。
基本的な構造は完成したのですが、どうしても解決できない部分があります。
辞書で同じ年齢(キー)に名前(値)を追加しようとすると、strはappendメソッドを持たないという旨のエラーが出て来ます。そもそもなぜstrになっているのかよくわかりません。

def is_int(arg):
    """
    int型かどうか調べる関数
    """
    try:
        int(arg)
        return True

    except ValueError:
        return False

def DataBase():
    # 名前を値v, 年齢をキーkとする辞書型変数 {年齢:年齢}
    data_dict = {}
    dammy = 0
    while(True):
        # 年齢順にソートする
        data_dict = dict(sorted(data_dict.items()))
        if(dammy != 0):
            for k, v in data_dict.items():
                print("年齢:{} 名前:{}".format(k, v))
            msg = input("表示したい方の名前あるいは年齢を入力してください:")

            # 年齢を入力された場合
            if(is_int(msg)):
                for k, v in data_dict.items():
                    if(k == msg):
                        print("年齢:{} 名前:{}".format(k, v))

        # 名前を入力された場合
            else:
                for k, v in data_dict.items():
                    if(v == msg):
                        print("年齢:{} 名前:{}".format(k, v))

        while(True):
            dammy = 1
            opt = input("データを追加(add),削除(del),何もしない(continue)?:")
            if(opt == "continue"):
                break
            elif(opt == "add"):
                age = input("追加したい方の年齢を入力してください:")
                name = input("追加したい方の名前を入力してください:")
                if((is_int(age) and not is_int(name))):
                    if(age in data_dict.keys()):
                        data_dict[age].append(name)
                    else:
                        data_dict[age] = name
                    break
            elif(opt == "del"):
                name = input("削除したい方の名前を入力してください:")
                age = input("削除したい方の年齢を入力してください:")
                if((is_int(age) and not is_int(name))):
                    for k, v in data_dict.items():
                        if((k == age) and (v == name)):
                            data_dict.pop(k, v)
                            break
            else:
                print("yかnを入力してください")


if __name__ == "__main__":
    DataBase()

Yuji-no-MacBook-Air:tmp_file YujiNarita$ python hw0423.py 
データを追加(add),削除(del),何もしない(continue)?:add
追加したい方の年齢を入力してください:21
追加したい方の名前を入力してください:yuji
年齢:21 名前:yuji
表示したい方の名前あるいは年齢を入力してください:21
年齢:21 名前:yuji
データを追加(add),削除(del),何もしない(continue)?:add
追加したい方の年齢を入力してください:21
追加したい方の名前を入力してください:aya
Traceback (most recent call last):
  File "hw0423.py", line 181, in <module>
    DataBase()
  File "hw0423.py", line 164, in DataBase
    data_dict[age].append(name)
AttributeError: 'str' object has no attribute 'append'
Yuji-no-MacBook-Air:tmp_file YujiNarita$ 

Answer 1

data_dict[age] = nameをdata_dict[age] = [name]にすればその部分のエラーは出なくなると思います。delの部分がどう影響を受けるかは未検討です。

Answer 2

とりあえずバグ自体は次のように解消できます。

if((is_int(age) and not is_int(name))):
    if(age in data_dict.keys()):
        data_dict[age].append(name)
    else:
        data_dict[age] = [name]   # <= 文字列じゃなくて『文字列のリスト』を代入しなきゃ

collections.defaultdictを使った方がもっとすっきり書けるかもしれませんね。

Comments

[north_redwings]

delの際に消しても残ってしまいました...

[LouiS0616]

delするとき、for k, v in data_dict.items(): のvがリストになりますから、これに関する対処も考えないといけませんね。

[north_redwings]

なるほど...ちょっと直して来ます!

[north_redwings]

できました!
いつもありがとうございます...