u++の備忘録

言語処理100本ノック 2020「05. n-gram」

問題文

nlp100.github.io

問題の概要

n-gram*1を作る関数を作成します。

def n_gram(target, n):
    return [target[idx:idx + n] for idx in range(len(target) - n + 1)]


text = 'I am an NLPer'
for i in range(1, 4):
    print(n_gram(text, i))
    print(n_gram(text.split(' '), i))