Pandas GroupBy — показывать только группы с более чем одним уникальным значением функции. ⇐ Python

[Anonymous]

У меня есть DataFrame df_things, который выглядит так, и я хочу спрогнозировать качество классификации перед обучением

Код: Выделить всё

A    B     C      CLASS
-----------------------
al1  bal1  cal1   Ship
al1  bal1  cal1   Ship
al1  bal2  cal2   Ship
al2  bal2  cal2   Cow
al3  bal3  cal3   Car
al1  bal2  cal3   Car
al3  bal3  cal3   Car

Я хочу сгруппировать строки по классам, чтобы иметь представление о распределении функций. Я делаю это с помощью (например, столбца «B»),

Код: Выделить всё

df_B = df_things.groupby('CLASS').B.value_counts()

что дает мне результаты

Код: Выделить всё

CLASS  B
-------------
ship   bal1  2
bal2  1
cow    bal2  2
car    bal2  1
bal3  2

Я хочу визуализировать только группы, имеющие более одного значения, чтобы это выглядело следующим образом:

Код: Выделить всё

CLASS  B
-------------
ship   bal1  2
bal2  1
car    bal2  1
bal3  2

Я немного застрял, есть идеи?

Подробнее здесь: https://stackoverflow.com/questions/539 ... ture-value