数据帧按多值分组

Question

如果我有一个如下所示的数据帧：

In [1]: df = pd.DataFrame([('bird', 'Falconiformes', 389.0),
   ...:                    ('bird', 'Psittaciformes', 24.0),
   ...:                    ('mammal', 'Carnivora', 80.2),
   ...:                    ('mammal', 'Primates', np.nan),
   ...:                    ('mammal', 'Carnivora', 58),
                           ('reptile', 'Some Animal', 30],
   ...:                    
   ...:                   columns=('class', 'order', 'max_speed'))
   ...: 

In [2]: df
Out[2]: 
class    order           max_speed
bird     Falconiformes   389.0
bird     Psittaciformes  24.0
mammal   Carnivora       80.2
mammal   Primates        NaN
mammal   Carnivora       58.0
reptile  Some Animal.    30

我应该如何按类将数据帧分为两组，一组是鸟类和爬行动物，另一组是哺乳动物？本课程只包括鸟类、哺乳动物和爬行动物

对于正常分组，这是我的方法

gb = df.groupby('class')    
for x in gb.groups:
    dff = gb.get_group(x)

Answer 1

只需创建一个映射，并将其作为一个新列应用于数据帧