第五章-酒店智能推荐---智慧旅游

第1关：构建用户-酒店矩阵

import numpy as np

def create_user_hotel_matrix(users, items, data, hotel_id):
    '''
    构建用户-酒店矩阵
    :param users: 用户数量，类型为整数
    :param items: 酒店数量，类型为整数
    :param data: 原始数据，类型为DataFrame
    :param hotel_id: 酒店ID的列表，类型为列表
    :return: user_hotel_matrix
    '''
    user_hotel_matrix = np.zeros((users, items))
    for line in data.itertuples():
        #********* Begin *********#
        user_hotel_matrix[line[3], hotel_id.index(line[1])] = line[4]
        #********* End *********#
    return user_hotel_matrix

第2关：酒店智能推荐

import numpy as np

def recommend_hotel(A, userid):
    '''
    向用户id为userid的用户推荐3家酒店
    :param A: 已经更新好了的矩阵A
    :param userid: 待推荐的userid，类型为整数
    :return: recommend
    '''
    #********* Begin *********#
    # 对矩阵A中userid对应的行进行升序排序
    ranklist = np.argsort(A[userid])
    #********* End *********#
    recommend = ranklist[-1:-4:-1]
    return recommend[-1], recommend[-2], recommend[-3]