Столбец из фрейма данных pandas при вызове имеет keyerror - PullRequest
0 голосов
/ 18 октября 2019

Я создаю систему рекомендаций к фильмам с использованием совместной фильтрации и создал список из 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

Я пытаюсь извлечь заголовок столбца, но продолжаю получать ошибку ключа. Можно мне помочь, пожалуйста?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...