Python関数は1つの値しか返しません。

Question

nltkから生​​成された単語の配列を関数から戻したいと思います。この関数は1単語しか返しません。返信してコメントを外すと、5単語が返されます。

私はそれらを返却し、他の機能で使用したいと考えています。

file = open('Text/Walden.txt', 'r',encoding="utf8") 
walden = file.read() 
walden = walden.split() 





def makePairs(arr): 
    pairs = [] 
    for i in range(len(arr)): 
     if i < len(arr)-1: 
      temp = (arr[i], arr[i+1]) 
      pairs.append(temp) 
    return pairs 

def generate(cfd, word = 'the', num = 5): 
    for i in range(num): 
     arr = []          # make an array with the words shown by proper count 
     for j in cfd[word]: 
      for k in range(cfd[word][j]): 
       arr.append(j) 

     word = arr[int((len(arr))*random.random())] # choose the word randomly from the conditional distribution 
    print(word, end=' ') 
    return(word) 
      #return random.choices(arr, k=num) 


pairs = makePairs(walden) 
cfd = nltk.ConditionalFreqDist(pairs) 
generate(cfd) 

出力は次のようである：

little The print The 
had 
>>> 

または

But girl?"— print girl?"— 
cases 
>>> 

Answer 1

あなたはあまりにも遠くインデントあなたreturn文を持っています。戻るそれアウト1つのレベル：あなたはreturnをコメントアウトし、代わりにprintステートメントを使用する場合

def generate(cfd, word = 'the', num = 5): 
    for i in range(num): 
     arr = []          # make an array with the words shown by proper count 
     for j in cfd[word]: 
      for k in range(cfd[word][j]): 
       arr.append(j) 
     #print(word, end=' ') 
     word = arr[int((len(arr))*random.random())] # choose the word randomly from the conditional distribution 

    return(word) 

、printがfor i in range(num)...で反復ごとに呼び出されます。そのため、5つのプリント出力が得られます。

Answer 2

returnステートメントがforループの範囲内にあるため、1語だけを返します。その最初の反復を行い、wordをランダムに選んですぐに戻します。

これは、私はそれを修正したい方法です - これはまたrandom.choicesのためのきちんとした場所である：この行で

def generate(cfd, word = 'the', num = 5): 
    arr = []     
    for j in cfd[word]: # assuming your code to parse cfd is correct 
     for k in range(cfd[word][j]): 
      arr.append(j) 
    return random.choices(arr, k=num)