计算Python中2个单词的所有组合的出现频率

Question

这个问题已经在这里有了答案：

• 在Python 8答案 中将列表成对循环（当前，下一个）
• 如何计算列表中元素的频率？ 29个答案

我有一段文字。 我想计算2个单词的所有可能组合（2个单词必须彼此相邻），例如：

"I have 2 laptops, I have 2 chargers"

结果应为：

"I have": 2
"have 2": 2
"2 laptops": 1
"Laptops, I": (Dont count)
"2 chargers": 1

我尝试过Regex，但事实是它不会两次计算字符串

我用过： \\b[az]{1,20}\\b \\b[az]{1,20}\\b

文字： cold chain, energy storage device, industrial cooling system

它几乎可以工作，但不包含“存储设备”， cooling system等字眼，因为它已经进行了energy storage和industrial cooling

感谢您的建议

Answer 1

您可以使用zip来获取每两个单词的分组，然后使用Counter来获取频率

>>> from collections import Counter
>>> text = "I have 2 laptops, I have 2 chargers"
>>> words = text.split()

>>> d = {' '.join(words):n for words,n in Counter(zip(words, words[1:])).items() if not  words[0][-1]==(',')}
>>> print (d)
{'I have': 2, 'have 2': 2, '2 laptops,': 1, '2 chargers': 1}

>>> import json
>>> print (json.dumps(d, indent=4))
{
    "I have": 2,
    "have 2": 2,
    "2 I": 1,
    "2 chargers": 1
}