矩阵生成

import numpy as np
import pandas as pd

recommand_matrix_dianying = np.zeros((943,1682))
names = ['user_id','item_id','rating','timestamp']
df = pd.read_csv('u.data',sep='\t',names=names)
for i in df.itertuples():
    recommand_matrix_dianying[i[1]-1][i[2]-1] = i[3]
print(recommand_matrix_dianying)

posted on 2019-01-14 16:41  零度风格  阅读(152)  评论(0编辑  收藏  举报

导航