Посчитайте, сколько раз значение столбца появляется и добавьте столбец к фрейму данных с ним

Question

У меня есть датафрейм с 4 столбцами, один из которых - имена людей, а другой - занятие, которое они практиковали. Я хочу, чтобы перед каждым рядом отображалось количество раз, когда появляется эта комбинация. Все найденные мной способы подсчета меняют фрейм данных или уменьшают размер фрейма данных, выводя каждую комбинацию только один раз. Я хотел бы, чтобы датафрейм оставался прежним, только с еще одним столбцом с количеством раз, которое существует комбинация. Кто-нибудь знает способ?

Answer 1

groupby + size

Предполагается, что ваши столбцы группировщика 0 и 2:

df['combination_count'] = df.groupby([0, 2])[1].transform('size')

Чтобы переместить новый столбец вперед:

cols = df.columns.tolist()
cols.insert(0, cols.pop(cols.index('combination_count')))
df = df.reindex(columns=cols)