Групування даних за категорією в pandas
Робота з категоріальними даними в Python
Kasey Jones
Research Data Scientist
Основи .groupby(): розбиття даних
adult = pd.read_csv("data/adult.csv")
adult1 = adult[adult["Above/Below 50k"] == " <=50K"]
adult2 = adult[adult["Above/Below 50k"] == " >50K"]
замінюється на
groupby_object = adult.groupby(by=["Above/Below 50k"])
Основи .groupby(): застосування функції
groupby_object = adult.groupby(by=["Above/Below 50k"])
Застосуйте функцію:
groupby_object.mean()
Age fnlgwt Education Num Capital Gain ...
Above/Below 50k
<=50K 36.783738 190340.86517 9.595065 148.752468 ...
>50K 44.249841 188005.00000 11.611657 4006.142456 ...
В один рядок:
adult.groupby(by=["Above/Below 50k"]).mean()
Визначення стовпців
Варіант 1: запускає .sum() лише для двох стовпців.
adult.groupby(by=["Above/Below 50k"])['Age', 'Education Num'].sum()
Age Education Num
Above/Below 50k
<=50K 909294 237190
>50K 346963 91047
Варіант 2: запускає .sum() для всіх числових стовпців, а потім робить вибірку.
adult.groupby(by=["Above/Below 50k"]).sum()[['Age', 'Education Num']]
Варіант 1 кращий — особливо для великих наборів даних
Групування за кількома стовпцями
adult.groupby(by=["Above/Below 50k", "Marital Status"]).size()
Above/Below 50k Marital Status
<=50K Divorced 3980
Married-AF-spouse 13
Married-civ-spouse 8284
Married-spouse-absent 384
Never-married 10192
Separated 959
Widowed 908
>50K Divorced 463
Married-AF-spouse 10 <--- Лише 10 записів
Married-civ-spouse 6692
...
Попрактикуймося з .groupby()
Робота з категоріальними даними в Python
