Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement