n-gram関数の中身がわからない

Question

https://qiita.com/yamaru/items/6d66445dbd5e7cef8640
にあります、

1. n-gram

与えられたシーケンス（文字列やリストなど）からn-gramを作る関数を作成せよ．この関数を用い，”I am an NLPer”という文から単語bi-gram，文字bi-gramを得よ．
ここでは、[lst[i:] for i in range(n)]で、1要素目始まり、2要素目始まり、…、n要素目始まりの要素列のリストを作成し、それらをzip関数に渡して前から順番に要素を抽出することで、連続するn要素からなる組を作っています。
なお、[lst[i:] for i in range(n)]の結果は1つのリストであるため、前に*をつけることでリストを展開し、それぞれの要素列に分けた上でzip関数に渡す必要があります。

なのですが、

def ngram(n, lst):
  # ex.
  # [str[i:] for i in range(2)] -> ['I am an NLPer', ' am an NLPer']
  # zip(*[str[i:] for i in range(2)]) -> zip('I am an NLPer', ' am an NLPer')
  return list(zip(*[lst[i:] for i in range(n)]))

str = 'I am an NLPer'
words_bi_gram = ngram(2, str.split())
chars_bi_gram = ngram(2, str)

print('単語bi-gram:', words_bi_gram)
print('文字bi-gram:', chars_bi_gram)

#結果

単語bi-gram: [('am', 'an'), ('I', 'am'), ('an', 'NLPer')]
文字bi-gram: [('I', ' '), (' ', 'N'), ('e', 'r'), ('a', 'm'), (' ', 'a'), ('n', ' '), ('L', 'P'), ('m', ' '), ('P', 'e'), ('N', 'L'), ('a', 'n')]

となっているのですが、

[lst[i:] for i in range(n)]

で[1要素目始まり、2要素目始まり、3要素目始まり、...]
というリストが得られるのはわかるのですが、
なぜ最後nの数の中身になっている(n=2ならば、```
[('am', 'an'), ('I', 'am'), ('an', 'NLPer')]

zip の仕様なのか、*の仕様なのかはわからないのですが、
詳しく教えていただけませんでしょうか?

Answer 1

中身がn個になる理由は、zipにn個のリストを渡しているためです。

zipは渡したリストの各要素を順番に取り出して連結しますので、2つのリストを指定すれば中身は2個ずつになり、3つのリストを指定すれば中身は3個ずつになります。

list1=['a1','a2','a3']
list2=['b1','b2','b3']
list3=['c1','c2','c3']

print(list(zip(list1, list2)))  # 2つのリストを指定した場合
# -> [('a1', 'b1'), ('a2', 'b2'), ('a3', 'b3')]

print(list(zip(list1, list2, list3)))  # 3つのリストを指定した場合
# -> [('a1', 'b1', 'c1'), ('a2', 'b2', 'c2'), ('a3', 'b3', 'c3')]