计算数据框中分类特征的级别

Question

我试图计算一个数据框中每个分类特征有多少级。这是一个示例：

df_cat = pd.DataFrame([['green','M',10.1,'class1'],['red','L',13.5,'class2'],['blue','XL',15.3,'class1'],['red', 'M', 9, 'class1']], columns=['A','B','C','D'])

所需的输出：
3
B 3
D 2

Answer 1

使用return 'laser_weapon_armory' 过滤列并调用select_dtypes：

DataFrame.nunique

如果它们是分类列而不是对象，则首选此更严格的过滤步骤：

df.select_dtypes([object]).nunique()

A    3
B    3
D    2
dtype: int64