def next_word_list(w, f):
'''
>>> next_word_list('red', ['red','yellow','red','black','grey','red'])
['yellow', 'black']
>>> next_word_list('big', ['small','big'])
[]
>>> next_word_list('big' , ['big','small'])
['small']
>>> next_word_list('blue', ['green'])
[]
>>> next_word_list('blue', [])
[]
>>> next_word_list('red', ['red','red','red'])
['red', 'red']
'''
l = []
for i in range(len(f)-1): #treiem 1 perquè a la ultima posició no cal mirar ja que no pot haber més
paraules després
if f[i] == w:
l.append(f[i+1]) #si no posem el +1 s'afegiria la w
return l
if __name__ == "__main__":
import doctest
doctest.testmod(verbose=True)