Я создаю систему рекомендаций к фильмам с использованием совместной фильтрации и создал список из 10 наиболее рекомендуемых фильмов с фреймом данных pandas.
Вот как я его создал:
movie_titles=pd.DataFrame(movieID)
movie_titles.columns=['movieId', 'title', 'genres']
df=pd.merge(df, movie_titles, on='movieId')
ratings=pd.DataFrame(df.groupby('title')['rating'].mean())
ratings['num of ratings']=pd.DataFrame(df.groupby('title')['rating'].count())
moviemat=df.pivot_table(index='userId', columns='title', values='rating')
movie_user_ratings=moviemat[movielens_title]
similar_to_movie=moviemat.corrwith(movie_user_ratings)
corr_movie=pd.DataFrame(similar_to_movie, columns=['Correlation'])
corr_movie=corr_movie.join(ratings['num of ratings'])
collab_ratings=corr_movie[corr_movie['num of ratings']>100][corr_movie['Correlation']<1].sort_values('Correlation',ascending=False).head(10)
Мой вывод:
![enter image description here](https://i.stack.imgur.com/wjHRw.png)
Я пытаюсь извлечь заголовок столбца, но продолжаю получать ошибку ключа. Можно мне помочь, пожалуйста?