Python进阶系列之怎么写出pythonic的代码

使用 in/not in 检查key是否存在于字典中

判断某个key是否存在于字典中时,一般的初学者想到的方法是,先以列表的形式把字典所有的key返回,在判断该key是否存在于key列表中

d = {}
keys = d.keys()
for k in keys:
    if key == k:
        print(True)
        break

更具有pythonic的用法是使用in关键字来判断key是否存在于字典中

if key in d: 
    print(true)
else:
    print(false)

使用setdefault()初始化字典的键值

使用字典的时候经常会遇到这种应用场景:动态更新字典,像如下代码,如果key不在d中那么就添加它并把它对应的值初始化为[],然后把元素append到空列表中

d = {"age":17}
if "name" not in d:
    d["name"]  = []
d["name"].append("zhangsan")

上面代码太low,我们可以使用setdefault来实现更pythonic的写法

代码:
d = {"age":17}
print(d)
d.setdefault("name",[]).append("zhangsan")
print(d)
结果:
{"age":17}
{"age":17,"name":["zhangsan"]}

用get获取字典中的值

关于获取字典中的值，一种简单的方式就是用d[x]访问该元素，但是这种情况在 key 不存在的情况下会报 KeyError 错误，当然你可以先用 in 操作检查 key 是否在字典中再获取，不过这种方式不符合 Python 之禅中说的

print(d.get("name", "default"))

使用字典推导式

numbers = [1, 2, 3]
d = {number: number * 2 for number in numbers}