рейтинги фильмов индекс user_id

Question

Я пытаюсь создать систему рекомендаций по фильмам с использованием ratings.csv. Я построил матрицу оценок, используя pivot () по user_id и movie_id.

mydf = mydf.pivot(index='user_Id', columns='movie_Id', values='rating').fillna(0).astype(int)

mymatrix = mydf.as_matrix()

Answer 1

Просто сделайте это:

dict(zip(list(mydf.index), list(range(len(mydf.index)))))

{22: 0, 23: 1, 25: 2, 31: 3, 142: 4}

Вот полный код

user_Id;movie_Id;rating
22;453;0
23;545;1
25;642;2
31;237;3
142;348;4

import pandas as pd
mydf = pd.read_clipboard(sep=';')
mydf = mydf.pivot(index='user_Id', columns='movie_Id', values='rating').fillna(0).astype(int)


dict(zip(list(mydf.index), list(range(len(mydf.index)))))

{22: 0, 23: 1, 25: 2, 31: 3, 142: 4}