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

Question

我正在对一个数据框进行一些分析，其中一列是一个整数，其值为 0 或 1(某种 bool 值，但采用整数形式)。它看起来像这样:

  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。

Answer 1

好的，下面是我能够做到的:

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

因此，groupby 对象中的 keys() 方法已经存储了可用的键，并且可以直接访问它。