说,你有这个功能ngrams
:
def ngrams(text, n=2):
return [text[i:i+n] for i in range(len(text)-n+1)]
现在将其应用于列表的所有元素相当容易:
>>> sentences = ['I like u', 'u like me']
>>> processed = [ngrams(sentence, n=2) for sentence in sentences]
>>> processed
[['I ', ' l', 'li', 'ik', 'ke', 'e ', ' u'],
['u ', ' l', 'li', 'ik', 'ke', 'e ', ' m', 'me']]
所以这很容易。要对 ngram 进行编号,您可以构建嵌套的 for 循环,但它看起来不太好。
相反,我们可以使用一个技巧: collections.defaultdict
,如果它在第一次访问时不存在,它将创建一个新项目。我们将它与itertools.count()
返回一个可迭代计数器结合起来。魔术__next__
方法是一个可调用的,第一次调用时返回第一个数字,然后是第二个,依此类推。defaultdict
每个新项目都会调用一次此方法
from collections import defaultdict
from itertools import count
reverse_vocabulary = defaultdict(count().__next__)
numbered = [[reverse_vocabulary[ngram] for ngram in sentence]
for sentence in processed]
print(numbered)
# [[0, 1, 2, 3, 4, 5, 6], [7, 1, 2, 3, 4, 5, 8, 9]]
现在反向词汇与您想要的相反:
defaultdict(<...>, {' m': 8, ' u': 6, 'I ': 0, 'li': 2, 'u ': 7, 'e ': 5, 'ke': 4, 'ik': 3,
' l': 1, 'me': 9})
我们通过反转映射来制作它的普通字典:
vocabulary = {number: ngram for ngram, number in reverse_vocabulary.items()}
这导致词汇成为普通字典
{0: 'I ', 1: ' l', 2: 'li', 3: 'ik', 4: 'ke', 5: 'e ', 6: ' u', 7: 'u ', 8: ' m', 9: 'me'}