python 集合の表示

Question

pythonの質問です
"paraparaparadise"と"paragraph"に含まれる文字bi-gramの集合を，それぞれ, XとYとして求め，XとYの和集合，積集合，差集合を求めよ．さらに，'se'というbi-gramがXおよびYに含まれるかどうかを調べよ．

という問題に対して自分はまずN-gramを取得する関数を作成し,文字列をbi-gramにしたのですがなぜか実行結果が「和集合: set()」という形になってしまい集合が表示されません.
このコードをどう修正すればよいですか?
どなたか添削してほしいです.

ちなみにこの問題は言語処理100本ノックの第1章-06の問題です.

def get_n_gram(n, str):
    words = str.split()
    result = []
    for i, j in enumerate(words):
        if i + n > len(words):
            return result
        result.append(words[i:i+n])

str1 = "paraparaparadise"
str2 = "paragrah"
X = get_n_gram(2, str1)
Y = get_n_gram(2, str2)

print(f'和集合: {set(X) | set(Y)}')
print(f'積集合: {set(X) & set(Y)}')
print(f'差集合: {set(X) - set(Y)}')
print("se" in (set(X) & set(Y)))

和集合: set()
積集合: set()
差集合: set()
False

Answer 1

今回、単語がターゲットのようですから、文字列を空白で分割する words = str.split() はいらないのではないでしょうか。
Python の場合、文字列はそのまま iterate して1文字ずつ取り出せます。

例:

s = "Apple"
for char in s:
    print(char, end=" ")
# A p p l e

修正後のコード

def get_n_gram(n, words):
    result = []
    for i, j in enumerate(words):
        if i + n > len(words):
            return result
        result.append(words[i : i + n])


str1 = "paraparaparadise"
str2 = "paragrah"
X = get_n_gram(2, str1)
Y = get_n_gram(2, str2)
print(X)
print(Y)

print(f"和集合: {set(X) | set(Y)}")
print(f"積集合: {set(X) & set(Y)}")
print(f"差集合: {set(X) - set(Y)}")
print("se" in (set(X) & set(Y)))

n-gram を作る関数は内包記法で書くこともできます。

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

Comments

[00_x9925]

解答ありがとうございました
この関数は他のコードでも使っていたものをそのままリユースしていたので見落としていました
チェック不足ですね...

修正した結果うまく作動しました