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

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

使用select_dtypes过滤列并调用DataFrame.nunique:

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

A    3
B    3
D    2
dtype: int64

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

# Categorical column conversion.
df = df.astype(dict.fromkeys('AB', 'category'))

df.dtypes    
A    category
B    category
C     float64
D      object
dtype: object

df.select_dtypes([pd.Categorical]).nunique()
A    3
B    3
dtype: int64