Как добавить столбец для записи повторяющихся данных в фрейме данных?

Question

У меня есть такой фрейм данных:

   user_id order_id
0        a        1
1        a        2
2        a        3
3        b        4
4        c        5

Теперь я хочу добавить столбец, чтобы показать, есть ли у пользователя каждого заказа несколько заказов:

   user_id order_id repetitive
0        a        1          1
1        a        2          1
2        a        3          1
3        b        4          0
4        c        5          0

Поскольку a имеет три порядка, тег равен 1. Я знаю, что метод value_counts может вычислить результат, но он показывает результат только после groupby. Я хочу объединить его с источником данных. Как мне этого добиться?

Answer 1

Используйте groupby и transform, чтобы получить счет, сохраняя ту же структуру.

df['repetitive'] = df.groupby('user_id').transform('count').gt(1).astype(int)