Python: получить подсчет частоты на основе двух столбцов (переменных) в панде dataframe
Здравствуйте, у меня есть следующий датафрейм.
Group Size
Short Small
Short Small
Moderate Medium
Moderate Small
Tall Large
Я хочу посчитать частоту того, сколько раз одна и та же строка появляется в кадре данных.
Group Size Time
Short Small 2
Moderate Medium 1
Moderate Small 1
Tall Large 1
2 ответа
Решение
Вы можете использовать групповые size
:
In [11]: df.groupby(["Group", "Size"]).size()
Out[11]:
Group Size
Moderate Medium 1
Small 1
Short Small 2
Tall Large 1
dtype: int64
In [12]: df.groupby(["Group", "Size"]).size().reset_index(name="Time")
Out[12]:
Group Size Time
0 Moderate Medium 1
1 Moderate Small 1
2 Short Small 2
3 Tall Large 1
Вы также можете попробовать pd.crosstab()
Group Size
Short Small
Short Small
Moderate Medium
Moderate Small
Tall Large
pd.crosstab(df.Group,df.Size)
Size Large Medium Small
Group
Moderate 0 1 1
Short 0 0 2
Tall 1 0 0
РЕДАКТИРОВАТЬ: Для того, чтобы получить ваш выход на место
pd.crosstab(df.Group,df.Size).replace(0,np.nan).\
stack().reset_index().rename(columns={0:'Time'})
Out[591]:
Group Size Time
0 Moderate Medium 1.0
1 Moderate Small 1.0
2 Short Small 2.0
3 Tall Large 1.0
Другая возможность - использовать .pivot_table()
а также aggfunc='size'
df_solution = df.pivot_table(index=['Group','Size'], aggfunc='size')