获取分类变量的类别列表（Python Pandas）

Question

我有一个Pandas DataFrame，它的列代表分类变量。如何获得类别列表？我在列上尝试使用.values，但是不会返回唯一的级别。

谢谢！

Answer 1

您也可以使用value_counts（），但仅当将其与列名一起使用时，它才有效，同时您还将获得每个类别的计数。 示例：

dataframe ['列名'] .value_counts（）

或者，如果您想要变量中类别的总数，则可以执行此操作，

dataframe ['Columnn name']。value_counts（）。count（）

Answer 2

我认为需要Series.cat.categories或unique：

np.random.seed(1245)

a = ['No', 'Yes', 'Maybe']
df = pd.DataFrame(np.random.choice(a, size=(10, 3)), columns=['Col1','Col2','Col3'])
df['Col1'] = pd.Categorical(df['Col1'])

print (df.dtypes)
Col1    category
Col2      object
Col3      object
dtype: object

print (df['Col1'].cat.categories)
Index(['Maybe', 'No', 'Yes'], dtype='object')

print (df['Col2'].unique())
['Yes' 'Maybe' 'No']

print (df['Col1'].unique())
[Maybe, No, Yes]
Categories (3, object): [Maybe, No, Yes]

Answer 3

尝试执行以下代码。

List_Of_Categories_In_Column=list(df['Categorical Column Name'].value_counts().index)

Answer 4

df.column name.value_counts() # 查看列中每个类别的值总数

df.column name.value_counts().index # 仅查看类别名称

df.column name .value_counts().count() # 查看一列有多少个类别（仅数字）