这是在用pandas实现协同过滤(推荐系统),上面都有注释,输出df如下,但是如果再输出df[1]就会报错,提示不存在1这个关键词,'1'也不行,好奇哪里出了问题
df=pd.read_csv(DATA_PATH,usecols=range(3)) # 读取csv文件
df=df.pivot_table(index=['userId'],columns=['movieId'],values=['rating']) # 做透视表,用户-物品评分表
print(df)
print(df[1])
这是部分原本csv的数据集截图
输出df.columns如下
print(df.columns.values)如下