尝试通过索引访问pandas数据帧时出现KeyError:0

时间:2019-06-28 13:56:05

标签: python pandas dataframe

我目前正在处理一个小的玩具数据集

"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]时,出现键盘错误。有人知道我可能做错了吗?

0 个答案:

没有答案