微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何在GroupBy对象可能不包含某个键时,如何避免Pandas Groupby键错误

我正在对数据帧进行一些分析,其中一列是一个值为0或1的整数(布尔值排序,但是整数形式).它看起来像这样:

  Nat. | Result
-------|-------
  CA   |  1
  USA  |  0
  GB   |  1
  USA  |  1
  CA   |  0
  GB   |  1

我根据国籍列对数据进行了分组,其中一个值(上例中的GB)是由偶然产生的 – 一个所有成员都只有1的组.这产生了一个问题,因为我有一个函数,我称之为包含group_obj.get_group(0)的批次,这会导致运行时错误“KeyError:0”

我的问题:我想创建以下逻辑:

if (group_obj contains key 0):
   return group_obj.get_group(0)
else:
   print "Group Object contains no 0s"
   return null

谢谢

我使用的是Python2,Pandas和iPython Notebook.

解决方法:

好的,所以这就是我能够做到的:

if key1 in group_obj.groups.keys():
    #Do processing

因此,groupby对象中的keys()方法已经存储了可用的密钥,并且可以直接访问它.

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐