Проблема чтения таблицы sqlite в виде csv-файла в рекомендательной системе - PullRequest
Новая
       32

Проблема чтения таблицы sqlite в виде csv-файла в рекомендательной системе

0 голосов
/ 12 февраля 2019

Мне нужно импортировать таблицу sqlite для оценки в виде csv-файла для моей системы рекомендаций.Recommender.py выглядит следующим образом: 

import pandas as pd
import numpy as np
from .models import movies,rating

movies=pd.read_csv('movies.csv')
ratings=pd.read_csv('ratings.csv')

 movies.head()
 ratings.head()

 def replace_name(x):
 return movies[movies['movie_id']==x].title.values[0]

 ratings.movie_id=ratings.movie_id.map(replace_name)

 userRatings=ratings.pivot_table(index=['account_id'], columns=
 ['movie_id'], values='rating')
  corrMatrix=userRatings.corr(method='pearson')

 myRatings=userRatings.loc[5].dropna()
 simCandidates=pd.Series()
 for i in range (0,len(myRatings.index)):
     sims=corrMatrix[myRatings,index[i]].dropna()
     sims=sims.map(lambda x: x*myRatings[i])
     simCandidates=simCandidates.append(sims)

 simCandidates.sort.values(inplace=True,ascending=False)
 simCandidates=simCandidates.groupby(simCandidates.index).sum()
 simCandidates=simCandidates.drop(myRatings.index)

У меня есть базовое веб-приложение в django, которое получает оценки различных фильмов от пользователя.После того, как пользователь оценивает все фильмы, система рекомендаций рекомендует пользователю определенные фильмы.Я сохраняю данные оценок в модели rating .Тем не менее, я не могу прочитать таблицу базы данных непосредственно в файле Recommender.py, как показано выше, так как мне нужен файл CSV - 'rating.csv'.Как я могу напрямую импортировать таблицу рейтинга в моей базе данных sqlite в виде csv-файла в приведенном выше коде и какие изменения я должен внести в приведенный выше код?

Пожалуйста, помогите.

1 Ответ

0 голосов
/ 12 февраля 2019

Для чтения Sqlite в Pandas. 

import sqlite3
import pandas as pd
# Create your connection.
db = sqlite3.connect('rating.db')

df = pd.read_sql_query("SELECT * FROM table_name", db)

Чтобы использовать read_sql_table (), у вас должна быть установлена ​​необязательная зависимость SQLAlchemy.http://pandas.pydata.org/pandas-docs/stable/io.html#reading-tables

...