基于用户的协同电影推荐代码实现:
# -*- coding: UTF-8 -*-
'''
基于用户的推荐算法
'''
from math import sqrt,pow
import csv
import operator
def read_file():
#disfile = 'F:\download\数据集\电影\ml-latest-small\matings.csv'
with open("F:\download\数据集\电影\ml-latest-small\matings.csv", 'r', encoding="utf-8") as f:
csv_reader = csv.reader(f)
data_list = []
for row in csv_reader:
data_list.append(row)
#print(data_list[1])
list_datas=[[] for i in range(611)]
#print(list_datas)
for i in range(1,len(data_list)):
list_tag = data_list[i]
id = int(list_tag[0])
#print(id)
list_datas[id].append(list_tag[1])
list_datas[id].append(float(list_tag[2]))
#print(list_datas)
f={}
#f = { i: {} for i in range(1,len(list_datas))}