【发布时间】:2016-03-27 17:28:16
【问题描述】:
在 Graphlab 中,
我正在处理较大列表中的一小部分电影。
movieIds_5K_np = LL_features_SCD_min.to_numpy()[:,0]
ratings_33K_np = ratings_33K.to_numpy()
movieIds_5K_np 是一个包含我的电影 ID 的数组。 `ratings_33K_np' 是一个具有四列的数组,其第二列包含所有电影的电影 ID。
我只需要选择ratings_33K_np 中id 存在于`movieIds_5K_np' 中的行。
我尝试了这种方法,但它似乎不起作用:
ratings_5K_np = ratings_33K_np[ratings_33K_np[:,2]==movieIds_5K_np]
如何在 Graphlab 中或使用一些 Python 库来做到这一点?我应该说最初ratings_33K 和movieIds_5K 是作为SFrame 导入的。
谢谢
【问题讨论】: