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

@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']

Новые блоги и статьи Все статьи Все блоги /
Установка Qt Creator для C и C++: ставим среду, CMake и MinGW без фреймворка Qt 8Observer8 05.04.2026 Среду разработки Qt Creator можно установить без фреймворка Qt. Есть отдельный репозиторий для этой среды: https:/ / github. com/ qt-creator/ qt-creator, где можно скачать установщик, на вкладке Releases:. . .	AkelPad-скрипты, структуры, и немного лирики.. testuser2 05.04.2026 Такая программа, как AkelPad существует уже давно, и также давно существуют скрипты под нее. Тем не менее, прога живет, периодически что-то не спеша дополняется, улучшается. Что меня в первую очередь. . .	Отображение реквизитов в документе по условию и контроль их заполнения Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "ПланированиеСпецтехники", разработанного в конфигурации КА2. Данный документ берёт данные из другого нетипового документа. . .	Фото всей Земли с борта корабля Orion миссии Artemis II kumehtar 04.04.2026 Это первое подобное фото сделанное человеком за 50 лет. Снимок называют новым вариантом легендарной фотографии «The Blue Marble» 1972 года, сделанной с борта корабля «Аполлон-17». Новое фото. . .
Вывод диалогового окна перед закрытием, если документ не проведён Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: реализовать программный контроль на предмет проведения документа. . .	Программный контроль заполнения реквизитов табличной части документа Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: 1. Реализовать контроль заполнения реквизита. . .	wmic не является внутренней или внешней командой Maks 02.04.2026 Решение: DISM / Online / Add-Capability / CapabilityName:WMIC~~~~ Отсюда: https:/ / winitpro. ru/ index. php/ 2025/ 02/ 14/ komanda-wmic-ne-naydena/	Программная установка даты и запрет ее изменения Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: при создании документов установить период списания автоматически. . .