Medium_recsys_1.py

import pandas as pd
import numpy as np


data = pd.read_csv('movielens100k.csv')
data['userId'] = data['userId'].astype('string')
data['movieId'] = data['movieId'].astype('string')


users = list(set(data['userId'])) #list of all users
movies = list(set(data['movieId'])) #list of all movies

print(len(users))
print(len(movies))