python "二维" 字典

今天读在movielens数据集的推荐系统代码读到"二维"字典的写法，这里学习记录一下

def get_dict(records:list, type:str):#此处record为ratings表的前三项，userid movieid rating，通过上面一个函数分为训练和测试集
    # 先看看redis里面有没有,如果有直接返回type_dict
    if redis.exists(type+‘_dict‘):
        return pickle.loads(redis.get(type+‘_dict‘))
    #redis中没有的话，开始新建，记录依次为user_id movie_id rating
    user_movie = {} #新建一个字典
    for record in records:
        user_id = int(record[0])
        movie_id = int(record[1])
        rating = int(record[2])
        if user_id not in user_movie:
            user_movie[user_id]={}
        user_movie[user_id][movie_id] = rating #二维字典

• user_movie是一个空字典
• user_movie[user_id]这里新建了字典的第一维
• user_movie[user_id][movie_id]新建了字典的第二维

jupyter notebook中测试一下
#在定义的时候初始化二维字典
dict_2d = {‘a‘ : {‘a‘ : 1, ‘b‘ : 3}, ‘b‘ : {‘a‘ : 6}}
dict_2d[‘a‘][‘a‘]
1

#一步一步的创建上面这个“二维”字典
myDict = {}
myDict[‘a‘] = {}
myDict[‘a‘][‘a‘] = 1
myDict[‘a‘][‘b‘] = 3
myDict[‘b‘] = {}
myDict[‘b‘][‘a‘] = 6
#测试是不是每一个都读到了
print(myDict[‘a‘][‘a‘])
print(myDict[‘a‘][‘b‘])
print(myDict[‘b‘][‘a‘])
1
3
6

python "二维" 字典

原文：https://www.cnblogs.com/lonelyisland/p/14530925.html