Advertisement
FancyKing

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

Apr 28th, 2020
448
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.62 KB | None | 0 0
  1. import numpy as np
  2.  
  3. def create_user_hotel_matrix(users, items, data, hotel_id):
  4.     '''
  5.    构建用户-酒店矩阵
  6.    :param users: 用户数量,类型为整数
  7.    :param items: 酒店数量,类型为整数
  8.    :param data: 原始数据,类型为DataFrame
  9.    :param hotel_id: 酒店ID的列表,类型为列表
  10.    :return: user_hotel_matrix
  11.    '''
  12.     user_hotel_matrix = np.zeros((users, items))
  13.     for line in data.itertuples():
  14.         #********* Begin *********#
  15.         user_hotel_matrix[line[3], hotel_id.index(line[1])] = line[4]
  16.         #********* End *********#
  17.     return user_hotel_matrix
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement