随机读取10个文件并将数据保存为list写入txt文件,省略了神经网络训练的过程
for i in range(10):
rand_user = random.randint(1,6040)
m_id = []
m_score = []
result = []
for j in movie_info:
tmp = user(rand_user,j)
# print tmp
u_tmp = [j,tmp]
print u_tmp
result.append(u_tmp)
result.sort(key=lambda x:x[1],reverse=True)
f = open('data/'+str(rand_user)+'.txt','w')
for k in range(0,1000):
m_id.append(result[k][0])
m_score.append(result[k][1])
f.write(str(m_id)+'\n')
f.write(str(m_score))
f.close