python中字典setdefault方法使用

# 经检查，发现自己理解错了需求，请忽略当前博文

《像计算机科学家一样思考python》11.10练习中 练习 11-2

阅读字典方法setdefault的文档，并使用它来写一个更简洁的invert_dict

※words.txt的下载地址如下请点击这里

 

def invert_dict(n):
    # 返回一个列表，第一项用setdefault查找用户输入的单词n是否存在，第二项返回查到项值
    # 如果单词（键）不存在，则在字典中添加键并将值设为默认值（这里是None）
    fin = open('words.txt')
    words = dict()
    count = 1
    for line in fin:
        word = line.strip()
        words[word] = count
        count += 1
    return words.setdefault(n, None), words[n]

print(invert_dict('abc'))
print(invert_dict('apple'))

 输出结果：

(None, None)
(4452, 4452)