Python pandas: добавить столбец в мой dataframe, который считает переменную

У меня есть фрейм данных 'gt', подобный следующему:

org     group
org1      1
org2      1
org3      2
org4      3
org5      3
org6      3

и я хотел бы добавить столбец 'count' в gt dataframe, чтобы подсчитать число членов групп, ожидаемые результаты, подобные этому:

org     group   count
org1      1       2
org2      1       2
org3      2       1
org4      3       3
org5      3       3
org6      3       3

Я знаю, как это сделать для одного элемента группы, но не знаю, как повторить счет для всех элементов группы, вот код, который я использовал:

gtcounts = gt.groupby('group').count()

Кто-нибудь может помочь?

16 голосов | спросил UserYmY 22 AMpWed, 22 Apr 2015 11:28:32 +030028Wednesday 2015, 11:28:32

1 ответ


0

Позвоните transform это вернет серию, выровненную по оригинальному df:

In [223]:

df['count'] = df.groupby('group')['group'].transform('count')
df
Out[223]:
    org  group  count
0  org1      1      2
1  org2      1      2
2  org3      2      1
3  org4      3      3
4  org5      3      3
5  org6      3      3
ответил EdChum 22 AMpWed, 22 Apr 2015 11:35:54 +030035Wednesday 2015, 11:35:54

Похожие вопросы

Популярные теги

security × 330linux × 316macos × 2827 × 268performance × 244command-line × 241sql-server × 235joomla-3.x × 222java × 189c++ × 186windows × 180cisco × 168bash × 158c# × 142gmail × 139arduino-uno × 139javascript × 134ssh × 133seo × 132mysql × 132