Поиск подходящих элементов между строками (списком слов) серии Pandas и заданным списком

Question

У меня есть серия панд, где каждая строка представляет собой список слов. Пример серии

У меня есть список слов, скажем my_list = ['ball', 'apple']. Я хочу заменить все строки серии словами, которые появляются как в этой строке, так и в my_list . Какой самый быстрый способ сделать это? Можно ли использовать векторизацию для этого типа задач?

Я хочу, чтобы результат был следующим: Желаемый вывод

Answer 1

Попробуйте это:

df = pd.DataFrame({'col1':[['apple','ball','cat'],['cat','donkey'],['elephant','apple','rhino']]}, index=[0,1,2])

my_list = ['ball', 'apple']

pd.Series([[r for r in i if r in my_list] for i in df['col1']])

Вывод:

0    [apple, ball]
1               []
2          [apple]
dtype: object