ngramのコードの仕組みについてご教示頂きたいです。

Question

pythonの勉強をしています。
自然言語処理のNgramについてのコードを理解しようとしたのですが、躓いてしまったためご教示お願い致します。

def ngrams(seq, n):
    return [seq[i:i+n] for i in range(len(seq)-n+1)]

sent = ["I", "have", "an", "apple"]

print(ngrams(sent, 2))

# [['I', 'have'], ['have', 'an'], ['an', 'apple']]

＊上記のコードは引用させていただきました。
https://termina.io/posts/n-gram-explanation-and-implementation

自分の認識：
逆算的な考え方をしてみました。

出力結果になるようにするには
seq[i:i+n]の部分のインデックス値が

[0:1][1:2][2:3]

となる必要があると考えたのですが
forループを基準として考えた場合

• リストngramの引数は２で
• リストlenはリストの要素をカウントするため

for i in range(４-２+1)

つまり、iは３になり最初のインデックス値から
[3:5]
となるのではないかと考えました。

どうして異なるのかがわかりません。
ご教示いただけると幸いです。
よろしくお願い致します。

Answer 1

for i in range(4-2+1)

は

for i in range(3)

であり、

for i in [0, 1, 2]

です。そのため、iは0から始まり、2で終わります。
ですので、index値は

[0:2][1:3][2:4]

となり、正常に動きます。

Comments

[jury_new]

Mashiroさん、ご回答ありがとうございます。
range関数は繰り返し処理ということが理解しきれていませんでした。
この度はありがとうございました。

[jury_new]

[0:2][1:3][2:4]
もう一度考え直した結果、インデックス値にも自身の誤りを発見しました。

[Mashiro]

確かにそうですね。
左側のINDEXしか見てなくて、そのままコピーしたので、回答も間違ってますね。
修正しときます。

[jury_new]

ご丁寧にありがとうございます！