如何根据单词列表对字符串中的单词进行分组？

Question

我有一个单词和一个字符串的列表，我想创建一个新的列表，如果一个列表中的相同单词在字符串中，而字符串的下一个单词也在字符串中，它会附加它们，并将它们存储为列表中的新元素。你知道吗

keyword_list = ['individual', 'fixed', 'treatments', 'deposit', 'health',
                'millions', 'panic', 'decision', 'policy', 'insurance', 'account']

string1 = 'i want to buy individual insurance policy and you can get upto 2 millions for the cover do not panic i also want to open fixed deposit account'

new_list = ['individual insurance policy',
            'millions', 'panic', 'fixed deposit account']

Answer 1

您可以根据元素在keyword_list中的存在情况对它们进行分组，并使用" "加入这些组。你知道吗

>>> data = 'i want to buy individual insurance policy and you can get upto 2 millions for the cover do not panic i also want to open fixed deposit account'
>>> keyword_list = ['individual', 'fixed', 'treatments', 'deposit', 'health',
...                 'millions', 'panic', 'decision', 'policy', 'insurance', 'account']

现在，让我们将keyword_list转换为一个集合，这样查找会更快。你知道吗

>>> keys = set(keyword_list)

现在，让我们根据单词在data中的出现情况来对keys中的单词进行分组，如下所示

>>> from itertools import groupby
>>> [" ".join(grp) for res, grp in groupby(data.split(), keys.__contains__) if res]
['individual insurance policy', 'millions', 'panic', 'fixed deposit account']

对于传递给groupby的集合中的每个元素，在本例中它是data.split()，将调用keys.__contains__函数。并基于该函数调用的结果，形成组。因为我们只对存在于keys中的项目感兴趣，所以我们在列表理解中使用if res进行过滤。你知道吗