Найти строки в фрейме данных, которые должны содержать не менее 2 элементов из списка

В моем примере я возвращаю все строки, содержащие любой из элементов из list1. Я хочу быть более строгим и заставить его возвращать строки, содержащие как минимум два элемента из list1.

Это возможно?

import pandas as pd
data = [
    ['tom steve orange', 'jane'],
    ['dave smith green', 'fran'],
    ['brit dave red', 'terri']
]
cols = ['A', 'B']
df = pd.DataFrame(data, columns=cols)

list1 = ['dave', 'red', 'blue']

df = df[df['A'].str.contains('|'.join(list1))].reset_index(drop=True)
print(df)

текущий результат:

                  A      B
0  dave smith green   fran
1     brit dave red  terri

Желаемый результат:

                  A      B
0     brit dave red  terri

Первая строка все равно будет соответствовать, потому что в ней есть dave и blue

Code Different 22.04.2022 23:12

да, только что пересмотрел

ah2Bwise 22.04.2022 23:17
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения текстовых сообщений может быть настолько сложным или простым, насколько вы его сделаете. Как и в любом ML-проекте, вы можете выбрать...
7 лайфхаков для начинающих Python-программистов
7 лайфхаков для начинающих Python-программистов
В этой статье мы расскажем о хитростях и советах по Python, которые должны быть известны разработчику Python.
Установка Apache Cassandra на Mac OS
Установка Apache Cassandra на Mac OS
Это краткое руководство по установке Apache Cassandra.
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
В одном из недавних постов я рассказал о том, как я использую навыки количественных исследований, которые я совершенствую в рамках программы TPQ...
Создание персонального файлового хранилища
Создание персонального файлового хранилища
Вы когда-нибудь хотели поделиться с кем-то файлом, но он содержал конфиденциальную информацию? Многие думают, что электронная почта безопасна, но это...
Создание приборной панели для анализа данных на GCP - часть I
Создание приборной панели для анализа данных на GCP - часть I
Недавно я столкнулся с интересной бизнес-задачей - визуализацией сбоев в цепочке поставок лекарств, которую могут просматривать врачи и...
3
2
36
2
Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Ответ принят как подходящий

Вы можете использовать операции над множествами:

S = set(list1)

out = df[[len(set(l.split())&S)>=2 for l in df['A']]]

# or
# out = df[[len(S.intersection(l.split()))>=2 for l in df['A']]]

Выход:

                 A      B
2    brit dave red  terri

вы также можете сделать что-то вроде этого:

df = df[df['A'].str.findall(fr"\b(?:{'|'.join(list1)})\b").map(len)>=2]

'''
               A      B
2  brit dave red  terri

Другие вопросы по теме