从两个字符串列表中查找单词的交集

Question

I have two list of strings as:

a = ['ketab khaneh','danesh gah', 'shi rin i']

b = ['ketab khaneh','dan esh gah','shirin i']

I need the intersection of individual words from the list. For example, for above lists, my desired output is:

output = ['ketab','khaneh','gah','i']

How I can produce this output in python?

Answer 1

use set intersection method

a = " ".join(a).split(" ")
b = " ".join(b).split(" ")
output = list(set(a).intersection(b))

output

['ketab', 'khaneh', 'gah', 'i']

Answer 2

You can achieve this usingset() with itertools.chain() and map() as:

>>> from itertools import chain

>>> a=['ketab khaneh','danesh gah', 'shi rin i']
>>> b=['ketab khaneh','dan esh gah','shirin i']

>>> set(chain(*map(str.split, a))).intersection(chain(*map(str.split, b)))
set(['i', 'khaneh', 'ketab', 'gah'])