Python 学习笔记-字典

键的存在性

不推荐的:

dct.has_key(key)

Python的做法:

key in dct

Yoda 测试(The “Yoda Test”)

如果你非要Not

not key in dct

English,你会说么？:

key not in dct

字典取值

这你有一个dict和key,并且你想修改key对应的值，比如给它自增1。

比较逊一点的做法：:

if key not in dct:
    dct[key] = 0
dct[key] = dct[key] + 1

漂亮的做法:

dct[key] = dct.get(key, 0) + 1

关于 dct.get(key[, default]) ,如果 key 存在,返回 dct[key],否则返回 default

更好的做法是:

如果使用Python 2.7,可以用Counter来计数:

>>> from collections import Counter
>>> d = [1, 1, 1, 2, 2, 3, 1, 1]
>>> Counter(d)
Counter({1: 5, 2: 2, 3: 1})

更完整的例子:

>>> counter = Counter()
... for _ in range(10):
...     num = int(raw_input("Enter a number: "))
...     counter.update([num])
...
... for key, value in counter.iteritems():
...     print "You have entered {}, {} times!".format(key, value)
Enter a number: 1
Enter a number: 1
Enter a number: 2
Enter a number: 3
Enter a number: 51
Enter a number: 1
Enter a number: 1
Enter a number: 1
Enter a number: 2
Enter a number: 3
You have entered 1, 5 times!
You have entered 2, 2 times!
You have entered 3, 2 times!
You have entered 51, 1 times!

修改字典的值

有时字典里是经常修改的对象，需要初始化一些数据到这个字典，也需要修改其中的一些值。比如说，你在维护一个这样的dict:它的值都是链表。

写一个实现就是:

dct = {}
for (key, value) in data:
    if key in dct:
        dct[key].append(value)
    else:
        dct[key] = [value]

更Pythonic的做法是:

dct = {}
for (key, value) in data:
    group = dct.setdefault(key, []) # key might exist already
    group.append(value)

setdefault(key, default) 所做的是：如果存在，返回 dct[key] , 不存在则把 dct[key] 设为 default 并返回它。当一个默认的值是一个你可以修改的对象的时候这是很有用的。

使用 defaultdict

dct = defaultdict(list)
for (key, value) in data:
    dct[key].append(value) # all keys have a default already

defaultdict 非常棒, 它每生成一对新的 key-value ，就会给value一个默认值, 这个默认值就是 defaultdict 的参数。(注:defaultdict在模块 collections 中)