我目前正在处理一个小的玩具数据集
"user_id","object_id"
1,1
1,2
2,4
2,2
3,5
3,2
4,1
4,4
5,5
5,2
我想做的是创建一个表,该表指示用户是否与特定对象相关联。为此,我有此方法
def is_associated(df, target_object):
a = (df.object_id == target_object)
a = a.to_frame()
a = a.join(df.user_id)
a = a.groupby(["user_id"],as_index=False ).any()
return a
虽然这不是最优雅的方法,但它成功返回了以下数据帧。例如,如果目标对象是1,我得到
user_id object_id
0 1 True
1 2 False
2 3 False
3 4 True
4 5 False
但是,当我尝试调用a [0]时,出现键盘错误。有人知道我可能做错了吗?