正文
:[
4
,
1
,
3
,
4
,
5
],
'doreen'
:[
3
,
4
,
2
,
5
,
3
]
}
def
user_data(user_dict):
for
name
in
user_dict:
user_info[name] = {u
'消失的爱人'
: user_dict[name][
0
]}
user_info[name][u
'霍比特人3'
] = user_dict[name][
1
]
user_info[name][u
'神去村'
] = user_dict[name][
2
]
user_info[name][u
'泰坦尼克号'
] = user_dict[name][
3
]
user_info[name][u
'这个杀手不太冷'
] = user_dict[name][
4
]
user_data(user_dict)
try
:
with open(
'user_data.txt'
,
'w'
) as data:
for
key
in
user_info:
data.write(key)
for
key2
in
user_info[key]:
data.write(
'\t'
)
data.write(key2)
data.write(
'\t'
)
data.write(
'\t'
)
data.write(str(user_info[key][key2]))
data.write(
'\n'
)
data.write(
'\n'
)
except
IOError as err:
print
(
'File error: '
+ str(err))
计算皮尔逊相关系数,找出兴趣相投的用户:(插入自己的数据)
-
from
math
import
sqrt
-
-
-
def
sim_pearson(prefs, p1, p2):
-
-
si = {}
-
for
item
in
prefs[p1]:
-
if
item
in
prefs[p2]:
-
si[item] =
1
-
-
-
if
len(si) ==
0
:
-
return
0
-
-