矩阵生成
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)