Коды категорий в Pandas
У категориального столбца есть
аксессор cat:
через него читают список меток
и целочисленные номера вместо
подписей. Атрибут codes
отдаёт ряд с такими номерами
для каждой строки.
Возьмём таблицу с городами
в категориальном виде и выведем
коды столбца city:
import pandas as pd
data = {
'city': [
'Moscow',
'Paris',
'Moscow',
'Paris',
'Berlin',
],
}
places = pd.DataFrame(data)
places['city'] = places['city'].astype(
'category'
)
print(places['city'].cat.codes) # выведет 0 1
# 1 2
# 2 1
# 3 2
# 4 0
# dtype: int8
Число в ячейке указывает позицию
метки в списке Categories:
Berlin получает 0,
Moscow - 1,
Paris - 2.
Создайте категориальный столбец
team со значениями
A, B, A
и выведите ряд его кодов.
Соберите таблицу с марками
Ford, Kia,
Ford в категориальном
столбце brand.
Покажите целые номера
для каждой строки.
Из таблицы с отделами
sales, it,
sales получите
ряд кодов столбца dept
и выведите его.