Получить список категорий категориальных переменных (Python Pandas)

1

У меня есть pandas DataFrame со столбцом, представляющим категориальную переменную. Как я могу получить список категорий? Я попытался.values в столбце, но это не возвращает уникальные уровни.

Спасибо!

  • 3
    pd.unique (имя_столбец)
Теги:
pandas
categorical-data

1 ответ

0
Лучший ответ

Я считаю, что вам нужна 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]
  • 1
    Большое спасибо! Используется .unique :)

Ещё вопросы

Сообщество Overcoder
Наверх
Меню