我正在尝试仅使用pandas编写一个简单的推荐系统,我遇到了过滤部分的问题。我想选择RatingCounts列大于我选择的值的所有行。这会返回一个数据帧一列填充了正确的布尔但我不能用这个选择索引我的数据它给我一个值错误,如标题中所述。这是截图
答案 0 :(得分:3)
selection
似乎是一个包含1列RatingCounts
的2D数据框架。
错误发生在when you pass a 2D indexer到DataFrame.loc
。
您可以通过将1D布尔索引器传递给DataFrame.loc
来避免错误:
final_data.loc[selection['RatingCounts']]