在python中，如何将每个单词映射到后面的单词列表？

# 创建一个空的字典
word_map = {}

# 输入一个句子
sentence = "I love coding in Python"

# 将句子拆分成单词列表
words = sentence.split()

# 遍历每个单词
for i in range(len(words)-1):
    word = words[i]
    next_word = words[i+1]
    
    # 如果字典中已经存在该单词，则将后面的单词添加到对应的列表中
    if word in word_map:
        word_map[word].append(next_word)
    # 如果字典中不存在该单词，则创建一个新的列表，并将后面的单词添加到列表中
    else:
        word_map[word] = [next_word]

# 打印每个单词映射的列表
for word, next_words in word_map.items():
    print(word, ":", next_words)

I : ['love']
love : ['coding']
coding : ['in']
in : ['Python']