Создать матрицу взаимодействия в Python

Question

У меня есть набор данных, содержащий два столбца, user_id и product_name. DataFrame выглядит следующим образом:

index user_id product_name
0     user1   A
1     user1   A
2     user2   A
3     user3   B

Я ищу способ преобразовать эту таблицу в матрицу взаимодействия для системы рекомендаций:

       A   B
user1  2   0
user2  1   0
user3  0   1

Answer 1

Вы можете сделать:

df.pivot_table(index='user_id', columns='product_name', aggfunc=len, fill_value=0)