Как получить пользователей у которых есть оба элемента?

@HitGirl · Регистрация: 08.10.2015

Студворк — интернет-сервис помощи студентам

Здравствуйте!
Есть датафрейм (pandas) со следующими колонками: userId, movieId, rating.

Даны два id фильмов, movie_id_1 и movie_id_2
Необходимо отфильтровать датафрейм так, чтобы в нём остались только пользователи у которых есть ОБА фильма.

Мой вариант:

Python
1
2
3
4
5
users_m1 = movie_data.loc[movie_data['movieId'] == movie_id_1,'userId']
users_m2 = movie_data.loc[movie_data['movieId'] == movie_id_2,'userId']
intersect_users = np.intersect1d(users_mj,users_mi)
mean_movie_1 = movie_data.loc[(movie_data['movieId'] == movie_id_1) & (movie_data['userId'].isin(intersect_users)),'rating'].mean()
mean_movie_2 = movie_data.loc[(movie_data['movieId'] == movie_id_2) & (movie_data['userId'].isin(intersect_users)),'rating'].mean()

Данный код выполняется очень медленно, особенно команда np.intersect1d.
Подскажите, пожалуйста, как эффективно находить пересечение пользователей в pandas?

@Smrn · 05.12.2022, 09:06

Смущают loc в первых двух строчках
Как вариант:

Python
1
2
3
4
5
6
tmp = movie_data[(movie_data["movieId"] == movie_id_1) | (movie_data["movieId"] == movie_id_2)]
tmp = tmp.drop_duplicates(subset=["userId", "movieId"])
tmp = tmp.groupby("userId", as_index=False)["movieId"].count()
tmp = tmp[tmp["movieId"]>1]
# tmp["userId"] - список с искомыми юзерами
movie_data = movie_data[movie_data["userId"].isin(list(tmp["userId"]))]

@HitGirl · 06.12.2022, 11:33 **[ТС]**

Python
1
intersect_users = movie_2[movie_2['userId'].isin(users_1)]['userId']

Новые блоги и статьи Все статьи Все блоги /
Отчёт о затраченных материалах за определенный период с макетом печатной формы Maks 21.04.2026 Отчёт из решения ниже размещён в конфигурации КА2. Задача: разработка отчёта по затраченным материалам за определённый период, с возможностью вывода печатной формы отчёта с шапкой и подвалом. В. . .	Отчёт о спецтехнике находящейся в ремонте Maks 20.04.2026 Отчёт из решения ниже размещен в конфигурации КА2. Задача: отобразить спецтехнику, которая на данный момент находится в ремонте. Есть нетиповой документ "Заявка на ремонт спецтехники" который. . .	Памятка для бота и "визитка" для читателей "Semantic Universe Layer (Слой семантической вселенной)" Hrethgir 19.04.2026 Сгенерировано для краткого описания по случаю сборки и компиляции скелета серверного приложения. И пусть после этого скажут, что статьи сгенерированные AI - туфта и не интересно. И это не реклама -. . .	Запрет удаления строк ТЧ документа при определённом условии Maks 19.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "Аккумуляторы", разработанного в конфигурации КА2. У данного документа есть ТЧ, в которой в зависимости от прав доступа. . .
Модель заражения группы наркоманов alhaos 17.04.2026 Условия задачи сформулированы тут Суть: - Группа наркоманов из 10 человек. - Только один инфицирован ВИЧ. - Колются одной иглой. - Колются раз в день. - Колются последовательно через. . .	Мысли в слух. Про "навсегда". kumehtar 16.04.2026 Подумалось тут, что наверное очень глупо использовать во всяких своих установках понятие "навсегда". Это очень сильное понятие, и я только начинаю понимать край его смысла, не смотря на то что давно. . .	My Business CRM MaGz GoLd 16.04.2026 Всем привет, недавно возникла потребность создать CRM, для личных нужд. Собственно программа предоставляет из себя базу данных клиентов, в которой можно фиксировать звонки, стадии сделки, а также. . .	Знаешь почему 90% людей редко бывают счастливыми? kumehtar 14.04.2026 Потому что они ждут. Ждут выходных, ждут отпуска, ждут удачного момента. . . а удачный момент так и не приходит.

Опции темы